Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 149 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 
from .llama import LlamaAWQForCausalLM
Casper Hansen's avatar
Support Falcon 7B+40B  
Casper Hansen committed
3
4 
from .opt import OptAWQForCausalLM
from .falcon import FalconAWQForCausalLM