DR-Rakshitha commited on
Commit
6c8e87d
·
1 Parent(s): 33c67dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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