Julien Blanchon commited on
Commit
d146b7a
·
1 Parent(s): 006a9e0
Files changed (1) hide show
  1. requirements.txt +25 -0
requirements.txt ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies
2
+ lpips>=0.1.4
3
+ matplotlib>=3.10.6
4
+ numpy>=2.2.6
5
+ pytorch-msssim>=1.0.0
6
+ scikit-image>=0.25.2
7
+ scipy>=1.15.3
8
+ pyyaml>=6.0.2
9
+ gradio>=4.0.0
10
+ huggingface_hub>=0.24.0
11
+ spaces>=0.28.0
12
+
13
+ # PyTorch with CUDA 12.4 support (install first for compatibility)
14
+ --extra-index-url https://download.pytorch.org/whl/cu124
15
+ torch==2.6.0
16
+ torchvision==0.21.0
17
+ torchmetrics>=1.8.2
18
+
19
+ # Pre-built wheels from HuggingFace (PyTorch 2.6.0 + CUDA 12.4)
20
+ # Note: These wheels are built for Python 3.13 (cp313)
21
+ # If using Python 3.10/3.11, rebuild wheels for cp310/cp311
22
+ fused_ssim @ https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/fused_ssim-0.0.1%2Bb4fd832pt2.6.0cu124-cp313-cp313-linux_x86_64.whl
23
+ gsplat @ https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp313-cp313-linux_x86_64.whl
24
+
25
+