Skip to content

GitLab

Commit c39ac59b authored by Casper's avatar Casper
Browse files

Import from awq

parent 7fbe9bbc
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
awq/__init__.py 0 → 100644
View file @ c39ac59b
from awq.models.auto import AutoAWQForCausalLM
\ No newline at end of file
awq/entry.py
View file @ c39ac59b
...@@ -4,7 +4,7 @@ import torch ...@@ -4,7 +4,7 @@ import torch
import argparse import argparse
from lm_eval import evaluator from lm_eval import evaluator
from transformers import AutoTokenizer 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_clip import apply_clip
from awq.quantize.auto_scale import apply_scale from awq.quantize.auto_scale import apply_scale
from awq.utils.lm_eval_adaptor import LMEvalAdaptor from awq.utils.lm_eval_adaptor import LMEvalAdaptor
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment