fix init

huggingface · Feb 6, 2024 · dde4ced · dde4ced
1 parent d025cb4
commit dde4ced
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 1 deletion.
diff --git a/docs/source/en/index.md b/docs/source/en/index.md
@@ -162,7 +162,7 @@ Flax), PyTorch, and/or TensorFlow.
 |                  [InstructBLIP](model_doc/instructblip)                  |       ✅        |         ❌         |      ❌      |
 |                       [Jukebox](model_doc/jukebox)                       |       ✅        |         ❌         |      ❌      |
 |                      [KOSMOS-2](model_doc/kosmos-2)                      |       ✅        |         ❌         |      ❌      |
-|                      [LagLlama](model_doc/lagllama)                      |       ❌        |         ❌         |      ❌      |
+|                      [LagLlama](model_doc/lagllama)                      |       ✅        |         ❌         |      ❌      |
 |                      [LayoutLM](model_doc/layoutlm)                      |       ✅        |         ✅         |      ❌      |
 |                    [LayoutLMv2](model_doc/layoutlmv2)                    |       ✅        |         ❌         |      ❌      |
 |                    [LayoutLMv3](model_doc/layoutlmv3)                    |       ✅        |         ✅         |      ❌      |

diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py
@@ -549,6 +549,7 @@
     "models.levit": ["LEVIT_PRETRAINED_CONFIG_ARCHIVE_MAP", "LevitConfig"],
     "models.lilt": ["LILT_PRETRAINED_CONFIG_ARCHIVE_MAP", "LiltConfig"],
     "models.llama": ["LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP", "LlamaConfig"],
+    "models.lagllama": ["LAGLLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP", "LagLlamaConfig"],
     "models.llava": [
         "LLAVA_PRETRAINED_CONFIG_ARCHIVE_MAP",
         "LlavaConfig",
@@ -2489,6 +2490,14 @@
             "LlamaPreTrainedModel",
         ]
     )
+    _import_structure["models.lagllama"].extend(
+        [
+            "LagLlamaForPrediction",
+            "LagLlamaForSequenceClassification",
+            "LagLlamaModel",
+            "LagLlamaPreTrainedModel",
+        ]
+    )
     _import_structure["models.llava"].extend(
         [
             "LLAVA_PRETRAINED_MODEL_ARCHIVE_LIST",
@@ -5262,6 +5271,7 @@
         Kosmos2Config,
         Kosmos2Processor,
     )
+    from .models.lagllama import LAGLLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP, LagLlamaConfig
     from .models.layoutlm import (
         LAYOUTLM_PRETRAINED_CONFIG_ARCHIVE_MAP,
         LayoutLMConfig,
@@ -6978,6 +6988,12 @@
             Kosmos2Model,
             Kosmos2PreTrainedModel,
         )
+        from .models.lagllama import (
+            LagLlamaForPrediction,
+            LagLlamaForSequenceClassification,
+            LagLlamaModel,
+            LagLlamaPreTrainedModel,
+        )
         from .models.layoutlm import (
             LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST,
             LayoutLMForMaskedLM,

diff --git a/src/transformers/utils/dummy_pt_objects.py b/src/transformers/utils/dummy_pt_objects.py
@@ -4454,6 +4454,34 @@ def __init__(self, *args, **kwargs):
         requires_backends(self, ["torch"])
 
 
+class LagLlamaForPrediction(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
+class LagLlamaForSequenceClassification(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
+class LagLlamaModel(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
+class LagLlamaPreTrainedModel(metaclass=DummyObject):
+    _backends = ["torch"]
+
+    def __init__(self, *args, **kwargs):
+        requires_backends(self, ["torch"])
+
+
 LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST = None