Jayant-Kernel commited on
fix: install unsloth_zoo and nest-asyncio properly
Browse files
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
|
| 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"],
|