alfonsovelp commited on
Commit
584cafa
·
verified ·
1 Parent(s): 8410227

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model_id = "mistralai/Mistral-7B-Instruct-v0.2"
13
  quantization_config = BitsAndBytesConfig(
14
  load_in_4bit=True,
15
  bnb_4bit_quant_type="nf4",
16
- bnb_4bit_compute_dtype="torch.float16",
17
  )
18
 
19
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=True, device_map="auto", token=hf_token)
 
13
  quantization_config = BitsAndBytesConfig(
14
  load_in_4bit=True,
15
  bnb_4bit_quant_type="nf4",
16
+ bnb_4bit_compute_dtype=torch.float1
17
  )
18
 
19
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=True, device_map="auto", token=hf_token)