invincible-jha commited on
Commit
f0b146e
·
1 Parent(s): 6f0fdff

Fix model initialization and add accelerate dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- gradio>=4.4.0
2
  transformers>=4.35.2
3
  torch>=2.1.1
4
  numpy>=1.24.3
@@ -17,4 +17,7 @@ pytest>=7.4.3
17
  crewai>=0.11.0
18
  huggingface_hub>=0.26.5
19
  psutil>=5.9.0
20
- python-json-logger>=2.0.7
 
 
 
 
1
+ gradio>=5.8.0
2
  transformers>=4.35.2
3
  torch>=2.1.1
4
  numpy>=1.24.3
 
17
  crewai>=0.11.0
18
  huggingface_hub>=0.26.5
19
  psutil>=5.9.0
20
+ python-json-logger>=2.0.7
21
+ accelerate>=0.26.0
22
+ bitsandbytes>=0.41.1
23
+ safetensors>=0.4.0