Create requirements.txt
Browse files- requirements.txt +6 -0
requirements.txt
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
accelerate==0.21.0
|
| 2 |
+
peft==0.4.0
|
| 3 |
+
bitsandbytes==0.40.2
|
| 4 |
+
transformers==4.33.0
|
| 5 |
+
trl==0.4.7
|
| 6 |
+
torch @ https://download.pytorch.org/whl/cu118/torch-2.1.0%2Bcu118-cp310-cp310-linux_x86_64.whl#sha256=a81b554184492005543ddc32e96469f9369d778dedd195d73bda9bed407d6589
|