Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -115,8 +115,8 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
|
|
115 |
model = AutoModelForCausalLM.from_pretrained(
|
116 |
MODEL_ID,
|
117 |
device_map="auto",
|
118 |
-
quantization_config=quantization_config,
|
119 |
-
attn_implementation="flash_attention_2",
|
120 |
)
|
121 |
|
122 |
# Create Gradio interface
|
|
|
115 |
model = AutoModelForCausalLM.from_pretrained(
|
116 |
MODEL_ID,
|
117 |
device_map="auto",
|
118 |
+
#quantization_config=quantization_config,
|
119 |
+
#attn_implementation="flash_attention_2",
|
120 |
)
|
121 |
|
122 |
# Create Gradio interface
|