Update environment.yml

environment.yml

@@ -6,30 +6,32 @@ channels:
 dependencies:
   - python=3.10
   - pip
-  - pytorch
-  - torchvision
-  - pytorch-cuda=12.1
-  - apex
-  - 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   
+  - 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
+      # O próprio Conda instalará torch, torchvision e apex a partir dos canais
+      # - torch
+      # - torchvision
+      # - apex