dvruette commited on
Commit
015750a
1 Parent(s): face2e4

update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -5
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
- torch==2.1.2
2
- transformers==4.37.2
3
- datasets==2.16.1
4
- accelerate==0.25.0
5
- safetensors==0.4.2
 
6
  concept-guidance @ git+https://github.com/dvruette/concept-guidance.git
 
1
+ torch~=2.1.2
2
+ transformers~=4.37.2
3
+ datasets~=2.16.1
4
+ accelerate~=0.25.0
5
+ safetensors~=0.4.2
6
+ bitsandbytes~=0.42.0
7
  concept-guidance @ git+https://github.com/dvruette/concept-guidance.git