Spaces:
Running
on
Zero
Running
on
Zero
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):
|