Maira-ghaffar commited on
Commit
5f7661a
·
verified ·
1 Parent(s): eb77d48

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -1
requirements.txt CHANGED
@@ -1,4 +1,21 @@
1
  sentence-transformers
2
  faiss-cpu
3
  numpy
4
- requests
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  sentence-transformers
2
  faiss-cpu
3
  numpy
4
+ requests
5
+
6
+ accelerate
7
+
8
+ then app.py mai
9
+ Replace this:
10
+ model = AutoModelForCausalLM.from_pretrained(
11
+ MODEL_ID,
12
+ config=config,
13
+ torch_dtype=torch.float32,
14
+ device_map="auto"
15
+ )
16
+ With this:
17
+ model = AutoModelForCausalLM.from_pretrained(
18
+ MODEL_ID,
19
+ config=config,
20
+ torch_dtype=torch.float32
21
+ ).to("cpu")