Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 108 Bytes
Newer Older
Casper Hansen's avatar
Refactor model class. Add LLaMa support (LLaMa-2, Vicuna, etc).  
Casper Hansen committed
1 
from .mpt import MptAWQForCausalLM
Casper Hansen's avatar
Add OPT support  
Casper Hansen committed
2
3 
from .llama import LlamaAWQForCausalLM
from .opt import OptAWQForCausalLM