Update handler.py

handler.py

@@ -2,13 +2,23 @@ import sys
 from subprocess import run
 
 
-run("pip install --upgrade pip", shell=True, check=True)
-run("pip uninstall torch -y", shell=True, check=True)
-run("pip uninstall torch -y", shell=True, check=True)
-run("pip install torch", shell=True, check=True)
+#run("pip install --upgrade pip", shell=True, check=True)
+#run("pip uninstall torch -y", shell=True, check=True)
+#run("pip uninstall torch -y", shell=True, check=True)
+#run("pip install torch", shell=True, check=True)
+
+#run("pip install unsloth unsloth_zoo --no-cache-dir --upgrade", shell=True, check=True)
+#run("pip uninstall unsloth unsloth_zoo torch -y", shell=True, check=True)
+
+run("pip install unsloth", shell=True, check=True)
+run("pip uninstall unsloth -y && pip install --upgrade --no-cache-dir --no-deps git+https://github.com/unslothai/unsloth.git", shell=True, check=True)
+run("pip install --no-deps bitsandbytes accelerate xformers==0.0.29 peft trl triton", shell=True, check=True)
+run("pip install --no-deps cut_cross_entropy unsloth_zoo", shell=True, check=True)
+run("pip install sentencepiece protobuf datasets huggingface_hub hf_transfer", shell=True, check=True)
+run("pip install --no-deps unsloth", shell=True, check=True)
+
+
 
-run("pip uninstall unsloth unsloth_zoo torch -y", shell=True, check=True)
-run("pip install unsloth unsloth_zoo --no-cache-dir --upgrade", shell=True, check=True)
 
 #run("pip install --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)
 #run("pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124", shell=True, check=True)