MWilinski commited on
Commit
d66d3b6
·
1 Parent(s): a0fb532
Files changed (1) hide show
  1. 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, torch_dtype=torch.bfloat16).model
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,