Spaces:
Sleeping
Sleeping
adjust feature for local model run
Browse files- requirements.txt +5 -2
requirements.txt
CHANGED
|
@@ -1,5 +1,8 @@
|
|
| 1 |
-
torch
|
| 2 |
transformers
|
| 3 |
-
unsloth
|
| 4 |
peft
|
| 5 |
gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
torch # installs the CUDA‐enabled build if your env supports it
|
| 2 |
transformers
|
|
|
|
| 3 |
peft
|
| 4 |
gradio
|
| 5 |
+
unsloth
|
| 6 |
+
bitsandbytes>=0.39.0 # CUDA‐enabled bitsandbytes wheel
|
| 7 |
+
|
| 8 |
+
#pip install bitsandbytes-0.39.0-cp310-cp310-linux_x86_64.whl
|