Update modeling_llava_qwen2.py
Browse files- modeling_llava_qwen2.py +1 -1
modeling_llava_qwen2.py
CHANGED
@@ -2332,4 +2332,4 @@ class LlavaQwen2ForCausalLM(Qwen2ForCausalLM, LlavaMetaForCausalLM):
|
|
2332 |
|
2333 |
|
2334 |
AutoConfig.register("llava-qwen2", LlavaQwen2Config)
|
2335 |
-
AutoModelForCausalLM.register(LlavaQwen2Config,
|
|
|
2332 |
|
2333 |
|
2334 |
AutoConfig.register("llava-qwen2", LlavaQwen2Config)
|
2335 |
+
AutoModelForCausalLM.register(LlavaQwen2Config, LlavaQwen2ForCausalLM)
|