nsfwalex commited on
Commit
f8b8471
·
1 Parent(s): accb937

update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -1
requirements.txt CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  pydantic==2.10.6
2
  insightface==0.7.3
3
  diffusers
@@ -8,7 +13,6 @@ einops
8
  opencv-python
9
  pyjwt
10
  torchsde
11
- torchvision==0.18.1
12
  compel
13
  onnxruntime-gpu
14
  hidiffusion
 
1
+ # --- make pip pull CUDA 12.1 wheels for PyTorch ---
2
+ --extra-index-url https://download.pytorch.org/whl/cu121
3
+
4
+ # Core CUDA stack (these three versions are a matched set)
5
+ spaces==0.40.0
6
  pydantic==2.10.6
7
  insightface==0.7.3
8
  diffusers
 
13
  opencv-python
14
  pyjwt
15
  torchsde
 
16
  compel
17
  onnxruntime-gpu
18
  hidiffusion