Spaces:
Running
on
A100
Running
on
A100
fix requirements.
Browse files- requirements.txt +6 -48
requirements.txt
CHANGED
@@ -1,48 +1,6 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
filelock==3.16.1 ; python_version >= "3.10" and python_version < "4.0"
|
8 |
-
fsspec==2024.9.0 ; python_version >= "3.10" and python_version < "4.0"
|
9 |
-
huggingface-hub==0.25.1 ; python_version >= "3.10" and python_version < "4.0"
|
10 |
-
idna==3.10 ; python_version >= "3.10" and python_version < "4.0"
|
11 |
-
importlib-metadata==8.5.0 ; python_version >= "3.10" and python_version < "4.0"
|
12 |
-
intel-openmp==2021.4.0 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Windows"
|
13 |
-
jinja2==3.1.4 ; python_version >= "3.10" and python_version < "4.0"
|
14 |
-
markupsafe==2.1.5 ; python_version >= "3.10" and python_version < "4.0"
|
15 |
-
mkl==2021.4.0 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Windows"
|
16 |
-
mpmath==1.3.0 ; python_version >= "3.10" and python_version < "4.0"
|
17 |
-
networkx==3.3 ; python_version >= "3.10" and python_version < "4.0"
|
18 |
-
numpy==2.1.1 ; python_version >= "3.10" and python_version < "4.0"
|
19 |
-
nvidia-cublas-cu12==12.1.3.1 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
20 |
-
nvidia-cuda-cupti-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
21 |
-
nvidia-cuda-nvrtc-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
22 |
-
nvidia-cuda-runtime-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
23 |
-
nvidia-cudnn-cu12==8.9.2.26 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
24 |
-
nvidia-cufft-cu12==11.0.2.54 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
25 |
-
nvidia-curand-cu12==10.3.2.106 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
26 |
-
nvidia-cusolver-cu12==11.4.5.107 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
27 |
-
nvidia-cusparse-cu12==12.1.0.106 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
28 |
-
nvidia-nccl-cu12==2.20.5 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
29 |
-
nvidia-nvjitlink-cu12==12.6.68 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
30 |
-
nvidia-nvtx-cu12==12.1.105 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version >= "3.10" and python_version < "4.0"
|
31 |
-
packaging==24.1 ; python_version >= "3.10" and python_version < "4.0"
|
32 |
-
pillow==10.4.0 ; python_version >= "3.10" and python_version < "4.0"
|
33 |
-
psutil==6.0.0 ; python_version >= "3.10" and python_version < "4.0"
|
34 |
-
pyyaml==6.0.2 ; python_version >= "3.10" and python_version < "4.0"
|
35 |
-
regex==2024.9.11 ; python_version >= "3.10" and python_version < "4.0"
|
36 |
-
requests==2.32.3 ; python_version >= "3.10" and python_version < "4.0"
|
37 |
-
safetensors==0.4.5 ; python_version >= "3.10" and python_version < "4.0"
|
38 |
-
sentencepiece==0.2.0 ; python_version >= "3.10" and python_version < "4.0"
|
39 |
-
sympy==1.13.3 ; python_version >= "3.10" and python_version < "4.0"
|
40 |
-
tbb==2021.13.1 ; python_version >= "3.10" and python_version < "4.0" and platform_system == "Windows"
|
41 |
-
tokenizers==0.19.1 ; python_version >= "3.10" and python_version < "4.0"
|
42 |
-
torch==2.3.0 ; python_version >= "3.10" and python_version < "4.0"
|
43 |
-
tqdm==4.66.5 ; python_version >= "3.10" and python_version < "4.0"
|
44 |
-
transformers==4.44.2 ; python_version >= "3.10" and python_version < "4.0"
|
45 |
-
triton==2.3.0 ; platform_system == "Linux" and platform_machine == "x86_64" and python_version < "3.12" and python_version >= "3.10"
|
46 |
-
typing-extensions==4.12.2 ; python_version >= "3.10" and python_version < "4.0"
|
47 |
-
urllib3==2.2.3 ; python_version >= "3.10" and python_version < "4.0"
|
48 |
-
zipp==3.20.2 ; python_version >= "3.10" and python_version < "4.0"
|
|
|
1 |
+
torch==2.3.0
|
2 |
+
diffusers==0.28.2
|
3 |
+
einops==0.6.1
|
4 |
+
transformers==4.44.2
|
5 |
+
sentencepiece==0.2.0
|
6 |
+
accelerate ==0.34.2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|