minhdang commited on
Commit
02b3b23
·
verified ·
1 Parent(s): cc9a9d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ nf4_config = BitsAndBytesConfig(
21
  bnb_4bit_use_double_quant=True,
22
  bnb_4bit_compute_dtype=torch.bfloat16
23
  )
24
- model_id = "MaziyarPanahi/Mistral-7B-Instruct-Aya-101"
25
  tokenizer = AutoTokenizer.from_pretrained(model_id)
26
  model = AutoModelForCausalLM.from_pretrained(model_id,
27
  # load_in_8bit=True,
 
21
  bnb_4bit_use_double_quant=True,
22
  bnb_4bit_compute_dtype=torch.bfloat16
23
  )
24
+ model_id = "Viet-Mistral/Vistral-7B-Chat"
25
  tokenizer = AutoTokenizer.from_pretrained(model_id)
26
  model = AutoModelForCausalLM.from_pretrained(model_id,
27
  # load_in_8bit=True,