FlawedLLM commited on
Commit
92b4405
·
verified ·
1 Parent(s): 6f18f2a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -5
requirements.txt CHANGED
@@ -1,6 +1,13 @@
1
  transformers==4.41.0
2
- accelerate==0.30.1
3
- bitsandbytes
4
- peft==0.11.1
5
- trl==0.8.6
6
- huggingface-hub==0.23.0
 
 
 
 
 
 
 
 
1
  transformers==4.41.0
2
+ # PyTorch and GPU Support
3
+ torch-cuda==12.1 # Or torch-cuda==11.8, depending on your exact needs
4
+ torchvision-cuda==0.13.1 # Ensure compatibility with PyTorch 12.1
5
+ cudatoolkit # No version specified; Conda will manage this
6
+ xformers # No version specified; Conda will manage this
7
+
8
+ # unsloth (and related)
9
+ unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
10
+ trl
11
+ peft
12
+ accelerate
13
+ bitsandbytes