Daemontatox commited on
Commit
38f56f2
·
verified ·
1 Parent(s): c37fa2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -68,9 +68,9 @@ h3 {
68
  def initialize_model():
69
  """Initialize the model with appropriate configurations"""
70
  quantization_config = BitsAndBytesConfig(
71
- load_in_4bit=True,
72
- bnb_4bit_compute_dtype=torch.bfloat16,
73
- bnb_4bit_use_double_quant=True
74
  )
75
 
76
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
 
68
  def initialize_model():
69
  """Initialize the model with appropriate configurations"""
70
  quantization_config = BitsAndBytesConfig(
71
+ load_in_8bit=True,
72
+ bnb_8bit_compute_dtype=torch.bfloat16,
73
+ bnb_8bit_use_double_quant=True
74
  )
75
 
76
  tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)