Spaces:
Runtime error
Runtime error
Commit
·
6c8e87d
1
Parent(s):
33c67dd
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,6 +64,7 @@ bnb_config = BitsAndBytesConfig(
|
|
| 64 |
bnb_4bit_quant_type=bnb_4bit_quant_type,
|
| 65 |
bnb_4bit_compute_dtype=compute_dtype,
|
| 66 |
bnb_4bit_use_double_quant=use_nested_quant,
|
|
|
|
| 67 |
)
|
| 68 |
|
| 69 |
# Remove GPU-specific check for bfloat16
|
|
|
|
| 64 |
bnb_4bit_quant_type=bnb_4bit_quant_type,
|
| 65 |
bnb_4bit_compute_dtype=compute_dtype,
|
| 66 |
bnb_4bit_use_double_quant=use_nested_quant,
|
| 67 |
+
bnb_4bit_disable_gpu=True, # Add this line to disable GPU quantization
|
| 68 |
)
|
| 69 |
|
| 70 |
# Remove GPU-specific check for bfloat16
|