Fuyu protection

huggingface · Nov 2, 2023 · b6c0c2b · b6c0c2b
1 parent fba863b
commit b6c0c2b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -343,7 +343,7 @@
  "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 @@
  _import_structure["models.efficientformer"].append("EfficientFormerImageProcessor")
  _import_structure["models.efficientnet"].append("EfficientNetImageProcessor")
  _import_structure["models.flava"].extend(["FlavaFeatureExtractor", "FlavaImageProcessor", "FlavaProcessor"])
- _import_structure["models.fuyu"].append("FuyuImageProcessor")
+ _import_structure["models.fuyu"].extend(["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 @@
  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 @@
  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