Jayant-Kernel commited on
Commit
2a3f319
·
unverified ·
1 Parent(s): 1670c46

fix: install unsloth_zoo and nest-asyncio properly

Browse files
Files changed (1) hide show
  1. train.py +1 -2
train.py CHANGED
@@ -2,8 +2,7 @@ import os, sys, pathlib, subprocess
2
 
3
  # Install deps
4
  subprocess.run([sys.executable, "-m", "pip", "install", "-q",
5
- "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git",
6
- "--no-deps"], check=True)
7
  subprocess.run([sys.executable, "-m", "pip", "install", "-q",
8
  "trl", "peft", "accelerate", "bitsandbytes", "wandb", "datasets",
9
  "git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git"],
 
2
 
3
  # Install deps
4
  subprocess.run([sys.executable, "-m", "pip", "install", "-q",
5
+ "unsloth", "unsloth_zoo", "nest-asyncio"], check=True)
 
6
  subprocess.run([sys.executable, "-m", "pip", "install", "-q",
7
  "trl", "peft", "accelerate", "bitsandbytes", "wandb", "datasets",
8
  "git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git"],