Update modeling_llava_qwen2.py
Browse files- modeling_llava_qwen2.py +3 -3
modeling_llava_qwen2.py
CHANGED
|
@@ -60,7 +60,7 @@ class SigLipImageProcessor:
|
|
| 60 |
return BatchFeature(data=data, tensor_type=return_tensors)
|
| 61 |
|
| 62 |
|
| 63 |
-
from
|
| 64 |
|
| 65 |
|
| 66 |
@dataclass
|
|
@@ -855,7 +855,7 @@ from transformers.utils import (
|
|
| 855 |
logging,
|
| 856 |
replace_return_docstrings,
|
| 857 |
)
|
| 858 |
-
from
|
| 859 |
|
| 860 |
|
| 861 |
if is_flash_attn_2_available():
|
|
@@ -2216,7 +2216,7 @@ class Qwen2ForSequenceClassification(Qwen2PreTrainedModel):
|
|
| 2216 |
)
|
| 2217 |
|
| 2218 |
|
| 2219 |
-
from
|
| 2220 |
|
| 2221 |
|
| 2222 |
class LlavaQwen2Model(LlavaMetaModel, Qwen2Model):
|
|
|
|
| 60 |
return BatchFeature(data=data, tensor_type=return_tensors)
|
| 61 |
|
| 62 |
|
| 63 |
+
from configuration_llava_qwen2 import SigLipVisionConfig
|
| 64 |
|
| 65 |
|
| 66 |
@dataclass
|
|
|
|
| 855 |
logging,
|
| 856 |
replace_return_docstrings,
|
| 857 |
)
|
| 858 |
+
from configuration_llava_qwen2 import Qwen2Config
|
| 859 |
|
| 860 |
|
| 861 |
if is_flash_attn_2_available():
|
|
|
|
| 2216 |
)
|
| 2217 |
|
| 2218 |
|
| 2219 |
+
from configuration_llava_qwen2 import LlavaQwen2Config
|
| 2220 |
|
| 2221 |
|
| 2222 |
class LlavaQwen2Model(LlavaMetaModel, Qwen2Model):
|