Skip to content

GitLab

Unverified Commit 84fb7e98 authored by Casper's avatar Casper Committed by GitHub
Browse files

Merge pull request #35 from casper-hansen/falcon-180b 

Support Falcon 180B
parents 9304af95 475481b4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
awq/models/auto.py
View file @ 84fb7e98
...@@ -8,6 +8,7 @@ AWQ_CAUSAL_LM_MODEL_MAP = { ...@@ -8,6 +8,7 @@ AWQ_CAUSAL_LM_MODEL_MAP = {
"opt": OptAWQForCausalLM, "opt": OptAWQForCausalLM,
"RefinedWeb": FalconAWQForCausalLM, "RefinedWeb": FalconAWQForCausalLM,
"RefinedWebModel": FalconAWQForCausalLM, "RefinedWebModel": FalconAWQForCausalLM,
"falcon": FalconAWQForCausalLM,
"bloom": BloomAWQForCausalLM, "bloom": BloomAWQForCausalLM,
"gptj": GPTJAWQForCausalLM "gptj": GPTJAWQForCausalLM
} }
......
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