File size: 2,118 Bytes
8a1012f
 
 
 
8dfcd19
8a1012f
 
 
 
 
 
 
 
 
 
ac23084
8a1012f
 
 
 
 
 
 
 
 
 
ac23084
8a1012f
 
 
 
 
 
 
 
 
ac23084
8a1012f
 
ac23084
8a1012f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
gradio
opencv-python-headless
pillow>=10.4.0
numpy==1.24.4
git+https://github.com/huggingface/diffusers.git@main

safetensors>=0.4.5
einops>=0.8.0
pyyaml>=6.0.2
omegaconf==2.3.0 
mediapy
librosa
moviepy
OmegaConf
beartype
sentencepiece
gradio[oauth]
mediapy
fastapi
uvicorn[standard]
tiktoken
transformers_stream_generator
rotary-embedding-torch
transformers
accelerate
safetensors
einops
decord
huggingface_hub
google-generativeai
gradio
tabulate
pydantic
soundfile
requests
hf_transfer
timm
ftfy
easydict
av


#vince
bitsandbytes==0.45.0
einop
numpy
rotary-embedding-torch
safetensors
sentencepiece
torch
torchvision
transformers



Common
                # Configuration system.
lpips>=0.1.4                    # LPIPS model for VAE training

# Dataloading
bson>=0.5.10                    # bson for Image dataloading
tensorflow>=2.16.1              # Video dataloading
opencv-python>=4.9.0.80         # OpenCV
sentencepiece>=0.2.0            # For Text Encoder

# Modeling
rotary-embedding-torch>=0.5.3   # Rotary positional embedding
transformers>=4.38.2            # Transformers
torchvision>=0.19.0             # Torchvision
tiktoken>=0.7.0                 # Tiktoken for generation
transformers_stream_generator>=0.0.5 #LLM generation support

# Metrics
torchmetrics>=1.3.2             # Core module for metric
pycocotools>=2.0.7              # COCO-related
torch-fidelity>=0.3.0           # FID-related

# Experiment Tracking
moviepy>=1.0.3                  # WandB Logging Image & Video
imageio>=2.34.0                 # WandB Logging Image & Video
tabulate>=0.9.0                 # Logging Table
deepdiff>=7.0.1                 # Find difference of config

# Testing
parameterized>=0.9.0            # Define multiple tests through decorators.

# Notebook
mediapy>=1.2.0                  # Notebook Visualization

# DevOPs
black >= 24             # Code formatting
flake8 >= 7              # Code style
isort >= 5               # Import sorting
pre-commit>=3.7.0               # Pre-commit hooks
expecttest>=0.2.1               # Pytorch dist tests
hypothesis>=6.100.1             # Fix randomness
av>=12.0.0