qnguyen3 commited on
Commit
a7e8634
·
verified ·
1 Parent(s): c26e9db

Update modeling_llava_qwen2.py

Browse files
Files changed (1) hide show
  1. 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, Qwen2ForCausalLM)
 
2332
 
2333
 
2334
  AutoConfig.register("llava-qwen2", LlavaQwen2Config)
2335
+ AutoModelForCausalLM.register(LlavaQwen2Config, LlavaQwen2ForCausalLM)