Daemontatox commited on
Commit
e704711
·
verified ·
1 Parent(s): 86ee556

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -140,7 +140,8 @@ def initialize_model():
140
  """Initialize the model with appropriate configurations"""
141
  quantization_config = BitsAndBytesConfig(
142
  load_in_4bit=True,
143
- bnb_4bit_compute_dtype=torch.float8,
 
144
  bnb_4bit_use_double_quant=True
145
  )
146
 
 
140
  """Initialize the model with appropriate configurations"""
141
  quantization_config = BitsAndBytesConfig(
142
  load_in_4bit=True,
143
+ bnb_4bit_compute_dtype=torch.bfloat16,
144
+ bnb_4bit_quant_type="nf4",
145
  bnb_4bit_use_double_quant=True
146
  )
147