leoeric commited on
Commit
d43b33a
·
1 Parent(s): ab19e03

Add requirements.txt for HF Spaces

Browse files
Files changed (1) hide show
  1. requirements.txt +33 -0
requirements.txt ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Requirements for Hugging Face Spaces
2
+ # PyTorch with CUDA support - HF Spaces automatically provides CUDA-enabled PyTorch
3
+ # When you select GPU hardware, PyTorch will have full CUDA support
4
+ torch>=2.0.0
5
+ torchvision>=0.15.0
6
+ # Note: CUDA toolkit is pre-installed on HF Spaces GPU instances
7
+ # You can verify with: torch.cuda.is_available() and torch.cuda.get_device_name(0)
8
+
9
+ # Core dependencies
10
+ transformers>=4.30.0
11
+ accelerate>=0.20.0
12
+ torchinfo
13
+ einops
14
+ scipy
15
+ sentencepiece
16
+ wandb[media]
17
+ torchmetrics[image]
18
+ simple_parsing
19
+ opencv-python
20
+ psutil
21
+ pyyaml
22
+ av==12.3.0
23
+
24
+ # Gradio for web interface
25
+ gradio>=4.44.1
26
+
27
+ # Git dependencies
28
+ git+https://github.com/KeKsBoTer/torch-dwt
29
+ git+https://github.com/huggingface/diffusers.git
30
+
31
+ # Note: decord and webdataset are optional and may not be needed for inference
32
+ # If needed, install via: pip install decord webdataset
33
+