Import from awq

c39ac59b · Casper · 7fbe9bbc · c39ac59b · c39ac59b
Commit c39ac59b authored Aug 27, 2023 by Casper
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

awq/__init__.py awq/__init__.py +1 -0

awq/entry.py awq/entry.py +1 -1

No files found.
--- a/awq/__init__.py
+++ b/awq/__init__.py
+from awq.models.auto import AutoAWQForCausalLM
\ No newline at end of file
--- a/awq/entry.py
+++ b/awq/entry.py
@@ -4,7 +4,7 @@ import torch
 import argparse
 from lm_eval import evaluator
 from transformers import AutoTokenizer
-from awq.models.auto import AutoAWQForCausalLM
+from awq import AutoAWQForCausalLM
 from awq.quantize.auto_clip import apply_clip
 from awq.quantize.auto_scale import apply_scale
 from awq.utils.lm_eval_adaptor import LMEvalAdaptor