MindVR commited on
Commit
090c9eb
·
verified ·
1 Parent(s): eb91bc9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -6
requirements.txt CHANGED
@@ -5,7 +5,7 @@ torch==2.2.2
5
  torchvision==0.17.2
6
  torchaudio==2.2.2
7
 
8
- # Install dependencies compatible with Unsloth v2024.4 and PyTorch 2.2.2 cu121
9
  xformers==0.0.25
10
  bitsandbytes==0.43.1
11
  accelerate==0.29.3
@@ -13,13 +13,11 @@ peft==0.10.0
13
  trl==0.17.0
14
  transformers==4.40.2
15
 
16
- # Install unsloth from the git repository using the specific commit hash for v2024.4
17
- # Using commit hash 037f3c3c45a5ef90530073862519c7f79e9c52ff which corresponds to tag v2024.4
18
- unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git@037f3c3c45a5ef90530073862519c7f79e9c52ff
19
 
20
  # Other packages
21
- # Removed --find-links for nightly builds as we are using a specific PyTorch index URL above
22
-
23
  sentencepiece
24
  protobuf
25
  datasets
 
5
  torchvision==0.17.2
6
  torchaudio==2.2.2
7
 
8
+ # Install dependencies compatible with Unsloth and PyTorch
9
  xformers==0.0.25
10
  bitsandbytes==0.43.1
11
  accelerate==0.29.3
 
13
  trl==0.17.0
14
  transformers==4.40.2
15
 
16
+ # Install unsloth from the git repository using the specific commit hash for v2024.4.1
17
+ # Using commit hash 7163265e20a585a167b8c5338110d8634d535223 which corresponds to tag v2024.4.1
18
+ unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git@7163265e20a585a167b8c5338110d8634d535223
19
 
20
  # Other packages
 
 
21
  sentencepiece
22
  protobuf
23
  datasets