Arjun Moorthy commited on
Commit
5984e38
·
1 Parent(s): 3fd805d

�� Add PEFT dependency for BioMistral model loading

Browse files
Files changed (1) hide show
  1. requirements.txt +14 -6
requirements.txt CHANGED
@@ -1,15 +1,23 @@
1
  # Medical Chatbot HF Space Requirements
2
 
3
  # Web framework
4
- gradio>=4.0.0
5
 
6
- # Machine learning libraries
7
- torch>=2.0.0
8
- transformers>=4.30.0
9
- accelerate>=0.20.0
10
 
11
  # HF Spaces GPU support
12
  spaces>=0.1.0
13
 
14
  # Basic utilities
15
- numpy>=1.21.0
 
 
 
 
 
 
 
 
 
1
  # Medical Chatbot HF Space Requirements
2
 
3
  # Web framework
4
+ gradio>=4.44.0
5
 
6
+ # Machine learning libraries - specific versions for compatibility
7
+ torch>=2.1.0,<3.0.0
8
+ transformers>=4.35.0,<5.0.0
9
+ accelerate>=0.24.0
10
 
11
  # HF Spaces GPU support
12
  spaces>=0.1.0
13
 
14
  # Basic utilities
15
+ numpy>=1.21.0,<2.0.0
16
+ requests>=2.28.0
17
+
18
+ # Additional dependencies for better device handling
19
+ safetensors>=0.4.0
20
+ tokenizers>=0.15.0
21
+
22
+ # PEFT for fine-tuned model loading
23
+ peft>=0.7.0,<1.0.0