md896 commited on
Commit
d118f9f
·
1 Parent(s): 6518b31

Fix TRL and Torchao dependency conflicts

Browse files
Files changed (1) hide show
  1. ultimate_sota_training.py +3 -1
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
- os.system("pip install trl accelerate wandb peft matplotlib -U")
 
 
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