EMsurya commited on
Commit
4a9f6e0
·
verified ·
1 Parent(s): 43ac03d
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
9
  model_id = "TheBloke/Mistral-7B-Instruct-v0.1-GGUF"
10
  model_file = "mistral-7b-instruct-v0.1.Q4_K_M.gguf"
11
  model_type = "mistral"
12
- quant_model = AutoModelForCausalLM.from_pretrained(model_id, model_file = model_file , model_type= model_type).to(device)
13
 
14
 
15
  def lechat_respond(
 
9
  model_id = "TheBloke/Mistral-7B-Instruct-v0.1-GGUF"
10
  model_file = "mistral-7b-instruct-v0.1.Q4_K_M.gguf"
11
  model_type = "mistral"
12
+ quant_model = AutoModelForCausalLM.from_pretrained(model_id, model_file = model_file , model_type= model_type)
13
 
14
 
15
  def lechat_respond(