Update env.yaml

env.yaml

@@ -5,16 +5,25 @@ channels:
   - conda-forge
   - defaults
 dependencies:
-  - python=3.9.16
+  - python=3.11.3
   - pytorch=2.0.1
   - pytorch-cuda=11.7
   - torchvision=0.15.2
-  - pip
   - pip:
+      - accelerate==0.19.0
+      - av==10.0.0
       - decord==0.6.0
-      - diffusers==0.15.0
-      - imageio==2.29.0
-      - transformers==4.29.2
+      - diffusers[torch]==0.16.0
+      - einops==0.6.1
+      - ffmpeg==1.4
+      - imageio==2.31.1
+      - imageio-ffmpeg==0.4.9
+      - pandas==2.0.1
+      - timm==0.6.13
+      - tqdm==4.65.0
+      - transformers==4.28.1
       - xformers==0.0.20
-      - einops
-      - omegaconf
+      - omegaconf==2.3.0
+      - natsort==8.4.0
+      - rotary_embedding_torch
+      - gradio==4.3.0