Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -2,10 +2,10 @@ import sys
|
|
| 2 |
from subprocess import run
|
| 3 |
|
| 4 |
|
| 5 |
-
#run("pip install unsloth", shell=True, check=True)
|
| 6 |
run("pip uninstall unsloth unsloth_zoo -y", shell=True, check=True)
|
| 7 |
run("pip install unsloth unsloth_zoo --no-cache-dir --upgrade", shell=True, check=True)
|
| 8 |
-
run("pip install
|
|
|
|
| 9 |
#run("pip uninstall unsloth -y && pip install --upgrade --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git", shell=True, check=True)
|
| 10 |
|
| 11 |
import torch
|
|
|
|
| 2 |
from subprocess import run
|
| 3 |
|
| 4 |
|
|
|
|
| 5 |
run("pip uninstall unsloth unsloth_zoo -y", shell=True, check=True)
|
| 6 |
run("pip install unsloth unsloth_zoo --no-cache-dir --upgrade", shell=True, check=True)
|
| 7 |
+
run("pip install --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)
|
| 8 |
+
#run("pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)
|
| 9 |
#run("pip uninstall unsloth -y && pip install --upgrade --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git", shell=True, check=True)
|
| 10 |
|
| 11 |
import torch
|