OP7 commited on
Commit
eae378d
·
verified ·
1 Parent(s): b09dd05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -32,10 +32,10 @@ import torch
32
 
33
  # Create a configuration for quantization
34
  quantization_config = BitsAndBytesConfig(
35
- load_in_8bit=True,
36
- bnb_8bit_compute_dtype="float16",
37
- bnb_8bit_use_double_quant=True,
38
- bnb_8bit_quant_type="nf4",
39
  )
40
 
41
  # Load the model and processor
 
32
 
33
  # Create a configuration for quantization
34
  quantization_config = BitsAndBytesConfig(
35
+ load_in_4bit=True,
36
+ bnb_4bit_compute_dtype="float16",
37
+ bnb_4bit_use_double_quant=True,
38
+ bnb_4bit_quant_type="nf4",
39
  )
40
 
41
  # Load the model and processor