Update app.py

app.py

@@ -1,39 +1,39 @@
 
-# # from huggingface_hub import login, HfFolder
-# import subprocess
+# from huggingface_hub import login, HfFolder
+import subprocess
 
-# subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
-# subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
+subprocess.run(["pip", "install", "--upgrade", "pip"], check=True)
+subprocess.run(["pip", "install", "--upgrade", "torch"], check=True)
 # subprocess.run(["pip", "install", "--upgrade", "transformers"], check=True)
-# subprocess.run(["pip", "install", "--upgrade", "bitsandbytes"], check=True)
+subprocess.run(["pip", "install", "--no-deps", "trl peft accelerate bitsandbytes"], check=True)
 # subprocess.run(["pip", "install", "--upgrade", "peft"], check=True)
-# subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
+subprocess.run(["pip", "install", "--upgrade", "xformers"], check=True)
 # subprocess.run(["pip", "install", "--upgrade", "accelerate"], check=True)
-# subprocess.run(["pip", "install", "unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git"], check=True)
+subprocess.run(["pip", "install", "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"], check=True)
 import subprocess
 
-def run_command(cmd, shell=False):
-    """Runs a shell command and prints the output."""
-    try:
-        result = subprocess.run(cmd, shell=shell, capture_output=True, text=True, check=True)
-        print(result.stdout)
-    except subprocess.CalledProcessError as e:
-        print(f"Error executing command: {e.stderr}")
+# def run_command(cmd, shell=False):
+#     """Runs a shell command and prints the output."""
+#     try:
+#         result = subprocess.run(cmd, shell=shell, capture_output=True, text=True, check=True)
+#         print(result.stdout)
+#     except subprocess.CalledProcessError as e:
+#         print(f"Error executing command: {e.stderr}")
 
-# 1. Create the conda environment
-run_command(["conda", "create", "-y", "--name", "unsloth_env", "python=3.10"])
+# # 1. Create the conda environment
+# run_command(["conda", "create", "-y", "--name", "unsloth_env", "python=3.10"])
 
-# 2. Activate the environment (Note: Requires shell=True)
-run_command("conda activate unsloth_env", shell=True)  
+# # 2. Activate the environment (Note: Requires shell=True)
+# run_command("conda activate unsloth_env", shell=True)  
 
-# 3. Install PyTorch and related packages with conda
-run_command("conda install pytorch-cuda=<12.1/11.8> pytorch cudatoolkit xformers -c pytorch -c nvidia -c xformers")
+# # 3. Install PyTorch and related packages with conda
+# run_command("conda install pytorch-cuda=<12.1/11.8> pytorch cudatoolkit xformers -c pytorch -c nvidia -c xformers")
 
-# 4. Install unsloth from the GitHub repository with pip
-run_command("pip install \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"")
+# # 4. Install unsloth from the GitHub repository with pip
+# run_command("pip install \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"")
 
-# 5. Install additional pip packages without dependencies
-run_command("pip install --no-deps trl peft accelerate bitsandbytes")
+# # 5. Install additional pip packages without dependencies
+# run_command("pip install --no-deps trl peft accelerate bitsandbytes")
 
 import os
 HF_TOKEN = os.environ["HF_TOKEN"]