Spaces:
Runtime error
Runtime error
Commit
•
638efdc
1
Parent(s):
cf864fd
Update requirements.txt
Browse files- requirements.txt +15 -2
requirements.txt
CHANGED
|
@@ -1,3 +1,16 @@
|
|
| 1 |
streamlit
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
streamlit
|
| 2 |
+
# Add specific version of cudatoolkit, pytorch-cuda, and pytorch
|
| 3 |
+
cudatoolkit
|
| 4 |
+
xformers
|
| 5 |
+
bitsandbytes
|
| 6 |
+
pytorch-cuda==11.8
|
| 7 |
+
pytorch
|
| 8 |
+
|
| 9 |
+
# Specify additional channels for cudatoolkit and pytorch
|
| 10 |
+
-c pytorch
|
| 11 |
+
-c nvidia
|
| 12 |
+
-c xformers
|
| 13 |
+
-c conda-forge
|
| 14 |
+
|
| 15 |
+
# Install unsloth with kaggle extras from GitHub repository
|
| 16 |
+
unsloth @ git+https://github.com/unslothai/unsloth.git
|