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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -140,7 +140,7 @@ 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.bfloat16,
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.float8,
144
  bnb_4bit_use_double_quant=True
145
  )
146