BidhanAcharya commited on
Commit
d8c73cf
·
verified ·
1 Parent(s): 8644f7f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
- # Install the required packages
2
- torch==2.0.1+cpu # CPU-only version of PyTorch
3
  transformers
4
  triton
5
  gradio
@@ -9,3 +8,4 @@ peft
9
  accelerate
10
  bitsandbytes
11
  git+https://github.com/unslothai/unsloth.git
 
 
1
+ torch==2.0.1 # General version for both CPU and GPU environments
 
2
  transformers
3
  triton
4
  gradio
 
8
  accelerate
9
  bitsandbytes
10
  git+https://github.com/unslothai/unsloth.git
11
+