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