AminFaraji commited on
Commit
7df4fe2
·
verified ·
1 Parent(s): 6e5bfb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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