Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
e9a6c72b
Unverified
Commit
e9a6c72b
authored
Nov 03, 2023
by
Lysandre Debut
Committed by
GitHub
Nov 03, 2023
Browse files
Fuyu protection (#27248)
parent
552ff244
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
11 additions
and
4 deletions
+11
-4
src/transformers/__init__.py
src/transformers/__init__.py
+4
-4
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 @
e9a6c72b
...
...
@@ -343,7 +343,7 @@ _import_structure = {
"models.focalnet"
:
[
"FOCALNET_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"FocalNetConfig"
],
"models.fsmt"
:
[
"FSMT_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"FSMTConfig"
,
"FSMTTokenizer"
],
"models.funnel"
:
[
"FUNNEL_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"FunnelConfig"
,
"FunnelTokenizer"
],
"models.fuyu"
:
[
"FUYU_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"FuyuConfig"
,
"FuyuProcessor"
],
"models.fuyu"
:
[
"FUYU_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"FuyuConfig"
],
"models.git"
:
[
"GIT_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"GitConfig"
,
"GitProcessor"
,
"GitVisionConfig"
],
"models.glpn"
:
[
"GLPN_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"GLPNConfig"
],
"models.gpt2"
:
[
"GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP"
,
"GPT2Config"
,
"GPT2Tokenizer"
],
...
...
@@ -987,7 +987,7 @@ else:
_import_structure
[
"models.efficientformer"
].
append
(
"EfficientFormerImageProcessor"
)
_import_structure
[
"models.efficientnet"
].
append
(
"EfficientNetImageProcessor"
)
_import_structure
[
"models.flava"
].
extend
([
"FlavaFeatureExtractor"
,
"FlavaImageProcessor"
,
"FlavaProcessor"
])
_import_structure
[
"models.fuyu"
].
app
end
(
"FuyuImageProcessor"
)
_import_structure
[
"models.fuyu"
].
ext
end
(
[
"FuyuImageProcessor"
,
"FuyuProcessor"
]
)
_import_structure
[
"models.glpn"
].
extend
([
"GLPNFeatureExtractor"
,
"GLPNImageProcessor"
])
_import_structure
[
"models.idefics"
].
extend
([
"IdeficsImageProcessor"
])
_import_structure
[
"models.imagegpt"
].
extend
([
"ImageGPTFeatureExtractor"
,
"ImageGPTImageProcessor"
])
...
...
@@ -4538,7 +4538,7 @@ if TYPE_CHECKING:
from
.models.focalnet
import
FOCALNET_PRETRAINED_CONFIG_ARCHIVE_MAP
,
FocalNetConfig
from
.models.fsmt
import
FSMT_PRETRAINED_CONFIG_ARCHIVE_MAP
,
FSMTConfig
,
FSMTTokenizer
from
.models.funnel
import
FUNNEL_PRETRAINED_CONFIG_ARCHIVE_MAP
,
FunnelConfig
,
FunnelTokenizer
from
.models.fuyu
import
FUYU_PRETRAINED_CONFIG_ARCHIVE_MAP
,
FuyuConfig
,
FuyuProcessor
from
.models.fuyu
import
FUYU_PRETRAINED_CONFIG_ARCHIVE_MAP
,
FuyuConfig
from
.models.git
import
GIT_PRETRAINED_CONFIG_ARCHIVE_MAP
,
GitConfig
,
GitProcessor
,
GitVisionConfig
from
.models.glpn
import
GLPN_PRETRAINED_CONFIG_ARCHIVE_MAP
,
GLPNConfig
from
.models.gpt2
import
GPT2_PRETRAINED_CONFIG_ARCHIVE_MAP
,
GPT2Config
,
GPT2Tokenizer
...
...
@@ -5117,7 +5117,7 @@ if TYPE_CHECKING:
from
.models.efficientformer
import
EfficientFormerImageProcessor
from
.models.efficientnet
import
EfficientNetImageProcessor
from
.models.flava
import
FlavaFeatureExtractor
,
FlavaImageProcessor
,
FlavaProcessor
from
.models.fuyu
import
FuyuImageProcessor
from
.models.fuyu
import
FuyuImageProcessor
,
FuyuProcessor
from
.models.glpn
import
GLPNFeatureExtractor
,
GLPNImageProcessor
from
.models.idefics
import
IdeficsImageProcessor
from
.models.imagegpt
import
ImageGPTFeatureExtractor
,
ImageGPTImageProcessor
...
...
src/transformers/utils/dummy_vision_objects.py
View file @
e9a6c72b
...
...
@@ -226,6 +226,13 @@ class FuyuImageProcessor(metaclass=DummyObject):
requires_backends
(
self
,
[
"vision"
])
class
FuyuProcessor
(
metaclass
=
DummyObject
):
_backends
=
[
"vision"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"vision"
])
class
GLPNFeatureExtractor
(
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