Spaces:
Running
on
Zero
Running
on
Zero
Daemontatox
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -68,9 +68,9 @@ h3 {
|
|
68 |
def initialize_model():
|
69 |
"""Initialize the model with appropriate configurations"""
|
70 |
quantization_config = BitsAndBytesConfig(
|
71 |
-
|
72 |
-
|
73 |
-
|
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)
|