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
|