feleanicusor commited on
Commit
bf9977d
·
verified ·
1 Parent(s): 2ec9b02

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,9 +15,10 @@ model = AutoModelForCausalLM.from_pretrained(
15
  load_in_4bit=True,
16
  bnb_4bit_compute_dtype=torch.bfloat16,
17
  bnb_4bit_quant_type="nf4",
 
18
  ),
19
  device_map="auto",
20
- torch_dtype=torch.bfloat16,
21
  )
22
  print("Model loaded!")
23
 
 
15
  load_in_4bit=True,
16
  bnb_4bit_compute_dtype=torch.bfloat16,
17
  bnb_4bit_quant_type="nf4",
18
+ bnb_4bit_use_double_quant=True,
19
  ),
20
  device_map="auto",
21
+ dtype=torch.bfloat16,
22
  )
23
  print("Model loaded!")
24