mistral-large-4-tiny-random / vllm_plugin.py
yujiepan's picture
Duplicate from tiny-random/mistral-large-4
57dbcb4
Raw History Blame Contribute Delete
193 Bytes
def register():
from vllm import ModelRegistry
ModelRegistry.register_model(
"MistralLarge4ForCausalLM",
"vllm_mistral_large4_tied:MistralLarge4TiedForCausalLM",
)