Spaces:
Paused
Paused
awq fix
Browse files- qa_engine/qa_engine.py +1 -1
qa_engine/qa_engine.py
CHANGED
@@ -66,7 +66,7 @@ class TransformersPipelineModel(LLM):
|
|
66 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
67 |
if "AWQ" in model_id:
|
68 |
model = AutoAWQForCausalLM.from_quantized(model_id, fuse_layers=True,
|
69 |
-
trust_remote_code=False, safetensors=True
|
70 |
else:
|
71 |
model = AutoModelForCausalLM.from_pretrained(
|
72 |
model_id,
|
|
|
66 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
67 |
if "AWQ" in model_id:
|
68 |
model = AutoAWQForCausalLM.from_quantized(model_id, fuse_layers=True,
|
69 |
+
trust_remote_code=False, safetensors=True).model
|
70 |
else:
|
71 |
model = AutoModelForCausalLM.from_pretrained(
|
72 |
model_id,
|