Update app.py

app.py

@@ -22,13 +22,26 @@ def run_command(cmd, shell=False):
     except subprocess.CalledProcessError as e:
         print(f"Error executing command: {e.stderr}")
 subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
+# subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
 # subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
+# Pip install command as a list
+pip_command = [
+    "pip", 
+    "install", 
+    "--upgrade", 
+    "--force-reinstall", 
+    "--no-cache-dir",
+    "torch==2.1.1",
+    "triton",
+    "--index-url", 
+    "https://download.pytorch.org/whl/cu121"
+]
+run_command(pip_command)
 run_command(["pip", "install", "--no-deps", "trl", "peft", "accelerate", "bitsandbytes"]) 
 # subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
-subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
+subprocess.run(["pip", "install", "xformers"], check=True)
 # subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
-subprocess.run(["pip", "install", "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"], check=True)
+subprocess.run(["unsloth[cu121-ampere-torch211] @ git+https://github.com/unslothai/unsloth.git"], check=True)
 # import subprocess