Update modeling_llava_qwen2.py
Browse files- modeling_llava_qwen2.py +1 -1
modeling_llava_qwen2.py
CHANGED
|
@@ -2226,7 +2226,7 @@ class LlavaQwen2Model(LlavaMetaModel, Qwen2Model):
|
|
| 2226 |
super(LlavaQwen2Model, self).__init__(config)
|
| 2227 |
|
| 2228 |
|
| 2229 |
-
class
|
| 2230 |
config_class = LlavaQwen2Config
|
| 2231 |
|
| 2232 |
def __init__(self, config):
|
|
|
|
| 2226 |
super(LlavaQwen2Model, self).__init__(config)
|
| 2227 |
|
| 2228 |
|
| 2229 |
+
class LlavaQwen2ForCausalLM(Qwen2ForCausalLM, LlavaMetaForCausalLM):
|
| 2230 |
config_class = LlavaQwen2Config
|
| 2231 |
|
| 2232 |
def __init__(self, config):
|