Spaces:
Running
on
Zero
Running
on
Zero
Daemontatox
commited on
Update app.py
Browse files
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.
|
|
|
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 |
|