aducsdr commited on
Commit
da3529e
·
verified ·
1 Parent(s): 296848f

Update requirements.txt

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