Update environment.yml

environment.yml

@@ -10,28 +10,26 @@ dependencies:
   - torchvision
   - pytorch-cuda=12.1
   - apex
-  - pip:
-      - einops
-      - omegaconf
-      - lpips
-      - bson
-      - tensorflow
-      - opencv-python
-      - sentencepiece
-      - diffusers
-      - rotary-embedding-torch
-      - transformers
-      - tiktoken
-      - transformers-stream-generator
-      - torchmetrics
-      - pycocotools
-      - torch-fidelity
-      - moviepy
-      - imageio
-      - tabulate
-      - deepdiff
-      - parameterized
-      - mediapy
-      - av
-      - gradio
-      - flash_attn==2.5.9.post1
+  - einops
+  - omegaconf
+  - lpips
+  - bson
+  - tensorflow
+  - opencv-python
+  - sentencepiece
+  - diffusers
+  - rotary-embedding-torch
+  - transformers
+  - tiktoken
+  - transformers-stream-generator
+  - torchmetrics
+  - pycocotools
+  - torch-fidelity
+  - moviepy
+  - imageio
+  - tabulate
+  - deepdiff
+  - parameterized
+  - mediapy
+  - av
+  - gradio