Spaces:
Running
Running
Fix TRL and Torchao dependency conflicts
Browse files
ultimate_sota_training.py
CHANGED
|
@@ -4,7 +4,9 @@
|
|
| 4 |
import os
|
| 5 |
print("📦 Installing State-of-the-Art Libraries (Unsloth & TRL)...")
|
| 6 |
os.system('pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"')
|
| 7 |
-
|
|
|
|
|
|
|
| 8 |
|
| 9 |
import httpx
|
| 10 |
import torch
|
|
|
|
| 4 |
import os
|
| 5 |
print("📦 Installing State-of-the-Art Libraries (Unsloth & TRL)...")
|
| 6 |
os.system('pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"')
|
| 7 |
+
# Removed the pip install -U line as Unsloth installs the correct versions of trl, accelerate, peft automatically
|
| 8 |
+
# Installing torchao separately since torch 2.5 has missing torch.int1 attribute in some versions of torchao. Actually unsloth handles torchao.
|
| 9 |
+
os.system("pip install wandb matplotlib")
|
| 10 |
|
| 11 |
import httpx
|
| 12 |
import torch
|