Update cognitivess_model/__init__.py

cognitivess_model/__init__.py

@@ -1,20 +1,116 @@
-from .configuration_cognitivess import CognitivessConfig
-from .modeling_cognitivess import CognitivessForCausalLM
-from .tokenization_cognitivess import CognitivessTokenizer
-from .tokenization_cognitivess_fast import CognitivessTokenizerFast
-
-from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer
-
-print("Registering CognitivessConfig...")
-AutoConfig.register("cognitivess", CognitivessConfig)
-print("CognitivessConfig registered successfully.")
-
-print("Registering CognitivessForCausalLM...")
-AutoModelForCausalLM.register(CognitivessConfig, CognitivessForCausalLM)
-print("CognitivessForCausalLM registered successfully.")
-
-print("Registering CognitivessTokenizer and CognitivessTokenizerFast...")
-AutoTokenizer.register(CognitivessConfig, 
-                       slow_tokenizer_class=CognitivessTokenizer, 
-                       fast_tokenizer_class=CognitivessTokenizerFast)
-print("CognitivessTokenizer and CognitivessTokenizerFast registered successfully.")
+# Copyright 2023 Cognitivess AI and The HuggingFace Inc. team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from typing import TYPE_CHECKING
+
+from ...utils import (
+    OptionalDependencyNotAvailable,
+    _LazyModule,
+    is_flax_available,
+    is_tf_available,
+    is_torch_available,
+)
+
+
+_import_structure = {
+    "configuration_Cognitivess": ["CognitivessConfig"],
+}
+
+
+try:
+    if not is_torch_available():
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    pass
+else:
+    _import_structure["modeling_Cognitivess"] = [
+        "CognitivessForCausalLM",
+        "CognitivessModel",
+        "CognitivessPreTrainedModel",
+        "CognitivessForSequenceClassification",
+        "CognitivessForTokenClassification",
+    ]
+
+try:
+    if not is_flax_available():
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    pass
+else:
+    _import_structure["modeling_flax_Cognitivess"] = [
+        "FlaxCognitivessForCausalLM",
+        "FlaxCognitivessModel",
+        "FlaxCognitivessPreTrainedModel",
+    ]
+
+try:
+    if not is_tf_available():
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    pass
+else:
+    _import_structure["modeling_tf_Cognitivess"] = [
+        "TFCognitivessModel",
+        "TFCognitivessForCausalLM",
+        "TFCognitivessForSequenceClassification",
+        "TFCognitivessPreTrainedModel",
+    ]
+
+
+if TYPE_CHECKING:
+    from .configuration_Cognitivess import CognitivessConfig
+
+    try:
+        if not is_torch_available():
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        pass
+    else:
+        from .modeling_Cognitivess import (
+            CognitivessForCausalLM,
+            CognitivessForSequenceClassification,
+            CognitivessForTokenClassification,
+            CognitivessModel,
+            CognitivessPreTrainedModel,
+        )
+
+    try:
+        if not is_flax_available():
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        pass
+    else:
+        from .modeling_flax_Cognitivess import (
+            FlaxCognitivessForCausalLM,
+            FlaxCognitivessModel,
+            FlaxCognitivessPreTrainedModel,
+        )
+
+    try:
+        if not is_tf_available():
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        pass
+    else:
+        from .modeling_tf_Cognitivess import (
+            TFCognitivessForCausalLM,
+            TFCognitivessForSequenceClassification,
+            TFCognitivessModel,
+            TFCognitivessPreTrainedModel,
+        )
+
+
+else:
+    import sys
+
+    sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)