Portx commited on
Commit
43aa402
·
verified ·
1 Parent(s): d914a4b

Update handler.py

Browse files
Files changed (1) hide show
  1. 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 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
 
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