Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
streamlit
|
| 2 |
-
torch==2.0.1
|
|
|
|
| 3 |
diffusers==0.21.4
|
| 4 |
transformers==4.35.0
|
| 5 |
huggingface-hub==0.18.0
|
|
|
|
| 1 |
streamlit
|
| 2 |
+
torch==2.0.1+cpu # use this instead of full CUDA version
|
| 3 |
+
-f https://download.pytorch.org/whl/torch_stable.html
|
| 4 |
diffusers==0.21.4
|
| 5 |
transformers==4.35.0
|
| 6 |
huggingface-hub==0.18.0
|