zoya-hammad commited on
Commit
552a4ed
·
1 Parent(s): 1d1ee9e

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
37
  if HF_TOKEN is None:
38
  raise ValueError("HF_TOKEN is not set. Check your .env file.")
39
  login(HF_TOKEN, add_to_git_credential=True)
40
- MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.3"
41
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
42
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
43
 
 
37
  if HF_TOKEN is None:
38
  raise ValueError("HF_TOKEN is not set. Check your .env file.")
39
  login(HF_TOKEN, add_to_git_credential=True)
40
+ MODEL_NAME = "mistralai/Mistral-7B-4bit"
41
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_TOKEN)
42
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto",use_auth_token=HF_TOKEN)
43