[Model] registry llama.py to vLLM
This commit is contained in:
@@ -78,6 +78,10 @@ def register_model():
|
|||||||
"SeedOssForCausalLM",
|
"SeedOssForCausalLM",
|
||||||
"vllm_kunlun.models.seed_oss:SeedOssForCausalLM")
|
"vllm_kunlun.models.seed_oss:SeedOssForCausalLM")
|
||||||
|
|
||||||
|
ModelRegistry.register_model(
|
||||||
|
"LlamaForCausalLM",
|
||||||
|
"vllm_kunlun.models.llama:LlamaForCausalLM")
|
||||||
|
|
||||||
|
|
||||||
def register_quant_method():
|
def register_quant_method():
|
||||||
"""to do"""
|
"""to do"""
|
||||||
|
|||||||
Reference in New Issue
Block a user