Vitrous commited on
Commit
2a75a89
·
verified ·
1 Parent(s): 469b97e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def mistral_model():
34
  print("CUDA is not available. CPU will be used.")
35
 
36
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
37
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2/model-00001-of-00003.safetensors")
38
 
39
  return model,tokenizer
40
 
 
34
  print("CUDA is not available. CPU will be used.")
35
 
36
  tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
37
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
38
 
39
  return model,tokenizer
40