aducsdr commited on
Commit
ed2f9cb
·
verified ·
1 Parent(s): 8ffd41a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -50
requirements.txt CHANGED
@@ -1,50 +1,5 @@
1
- # Critical PyTorch dependencies must come first
2
- torch==2.6.0
3
- torchvision==0.21.0
4
-
5
- # Common
6
- einops # Tensor operations
7
-
8
- # Training
9
- omegaconf # Configuration system.
10
- lpips # LPIPS model for VAE training
11
-
12
- # Dataloading
13
- bson # bson for Image dataloading
14
- tensorflow # Video dataloading
15
- opencv-python # OpenCV
16
- sentencepiece # For Text Encoder
17
-
18
- # Modeling
19
- diffusers # Diffusers
20
- rotary-embedding-torch # Rotary positional embedding
21
- transformers # Transformers
22
- tiktoken # Tiktoken for generation
23
- transformers_stream_generator #LLM generation support
24
-
25
- # Metrics
26
- torchmetrics # Core module for metric
27
- pycocotools # COCO-related
28
- torch-fidelity # FID-related
29
-
30
- # Experiment Tracking
31
- moviepy # WandB Logging Image & Video
32
- imageio # WandB Logging Image & Video
33
- tabulate # Logging Table
34
- deepdiff # Find difference of config
35
-
36
- # Testing
37
- parameterized # Define multiple tests through decorators.
38
-
39
- # Notebook
40
- mediapy # Notebook Visualization
41
-
42
- # DevOPs
43
- black # Code formatting
44
- flake8 # Code style
45
- isort # Import sorting
46
- pre-commit # Pre-commit hooks
47
- expecttest # Pytorch dist tests
48
- hypothesis # Fix randomness
49
- av # Process audio and video
50
- https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
 
1
+ # Dependências básicas. O resto será instalado via conda no app.py
2
+ gradio
3
+ einops
4
+ omegaconf
5
+ mediapy