Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
15c74a28
Unverified
Commit
15c74a28
authored
May 16, 2024
by
amyeroberts
Committed by
GitHub
May 16, 2024
Browse files
Fix VideoLlava imports (#30867)
* Fix VideoLlava imports * Update dummy objects
parent
4e17e7dc
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
9 additions
and
9 deletions
+9
-9
src/transformers/__init__.py
src/transformers/__init__.py
+2
-2
src/transformers/utils/dummy_pt_objects.py
src/transformers/utils/dummy_pt_objects.py
+0
-7
src/transformers/utils/dummy_vision_objects.py
src/transformers/utils/dummy_vision_objects.py
+7
-0
No files found.
src/transformers/__init__.py
View file @
15c74a28
...
...
@@ -1160,6 +1160,7 @@ else:
_import_structure
[
"models.swin2sr"
].
append
(
"Swin2SRImageProcessor"
)
_import_structure
[
"models.tvlt"
].
append
(
"TvltImageProcessor"
)
_import_structure
[
"models.tvp"
].
append
(
"TvpImageProcessor"
)
_import_structure
[
"models.video_llava"
].
append
(
"VideoLlavaImageProcessor"
)
_import_structure
[
"models.videomae"
].
extend
([
"VideoMAEFeatureExtractor"
,
"VideoMAEImageProcessor"
])
_import_structure
[
"models.vilt"
].
extend
([
"ViltFeatureExtractor"
,
"ViltImageProcessor"
,
"ViltProcessor"
])
_import_structure
[
"models.vit"
].
extend
([
"ViTFeatureExtractor"
,
"ViTImageProcessor"
])
...
...
@@ -3243,7 +3244,6 @@ else:
_import_structure
[
"models.video_llava"
].
extend
(
[
"VideoLlavaForConditionalGeneration"
,
"VideoLlavaImageProcessor"
,
"VideoLlavaPreTrainedModel"
,
"VideoLlavaProcessor"
,
]
...
...
@@ -5759,6 +5759,7 @@ if TYPE_CHECKING:
from
.models.swin2sr
import
Swin2SRImageProcessor
from
.models.tvlt
import
TvltImageProcessor
from
.models.tvp
import
TvpImageProcessor
from
.models.video_llava
import
VideoLlavaImageProcessor
from
.models.videomae
import
VideoMAEFeatureExtractor
,
VideoMAEImageProcessor
from
.models.vilt
import
ViltFeatureExtractor
,
ViltImageProcessor
,
ViltProcessor
from
.models.vit
import
ViTFeatureExtractor
,
ViTImageProcessor
...
...
@@ -7437,7 +7438,6 @@ if TYPE_CHECKING:
)
from
.models.video_llava
import
(
VideoLlavaForConditionalGeneration
,
VideoLlavaImageProcessor
,
VideoLlavaPreTrainedModel
,
VideoLlavaProcessor
,
)
...
...
src/transformers/utils/dummy_pt_objects.py
View file @
15c74a28
...
...
@@ -8417,13 +8417,6 @@ class VideoLlavaForConditionalGeneration(metaclass=DummyObject):
requires_backends
(
self
,
[
"torch"
])
class
VideoLlavaImageProcessor
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
])
class
VideoLlavaPreTrainedModel
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
]
...
...
src/transformers/utils/dummy_vision_objects.py
View file @
15c74a28
...
...
@@ -534,6 +534,13 @@ class TvpImageProcessor(metaclass=DummyObject):
requires_backends
(
self
,
[
"vision"
])
class
VideoLlavaImageProcessor
(
metaclass
=
DummyObject
):
_backends
=
[
"vision"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"vision"
])
class
VideoMAEFeatureExtractor
(
metaclass
=
DummyObject
):
_backends
=
[
"vision"
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment