Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -93,7 +93,7 @@ db = Chroma(persist_directory=CHROMA_PATH, embedding_function=embeddings)
|
|
93 |
MODEL_NAME = "tiiuae/falcon-7b-instruct"
|
94 |
|
95 |
model = AutoModelForCausalLM.from_pretrained(
|
96 |
-
MODEL_NAME, trust_remote_code=True,
|
97 |
)
|
98 |
model = model.eval()
|
99 |
|
|
|
93 |
MODEL_NAME = "tiiuae/falcon-7b-instruct"
|
94 |
|
95 |
model = AutoModelForCausalLM.from_pretrained(
|
96 |
+
MODEL_NAME, trust_remote_code=True, device_map="auto"
|
97 |
)
|
98 |
model = model.eval()
|
99 |
|