Daemontatox commited on
Commit
86ddf2a
·
verified ·
1 Parent(s): ab4f4a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -98,9 +98,9 @@ h3 {
98
  def initialize_model():
99
  """Initialize the model with appropriate configurations"""
100
  quantization_config = BitsAndBytesConfig(
101
- load_in_4bit=True,
102
- bnb_4bit_compute_dtype=torch.bfloat16,
103
- bnb_4bit_use_double_quant=True
104
  )
105
 
106
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
98
  def initialize_model():
99
  """Initialize the model with appropriate configurations"""
100
  quantization_config = BitsAndBytesConfig(
101
+ load_in_8bit=True,
102
+ bnb_8bit_compute_dtype=torch.bfloat16,
103
+ bnb_8bit_use_double_quant=True
104
  )
105
 
106
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)