Update the __all__

huggingface · Oct 1, 2024 · a7762a0 · a7762a0
1 parent 9c6f445
commit a7762a0
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/transformers/models/glm/modular_glm.py b/src/transformers/models/glm/modular_glm.py
@@ -204,3 +204,12 @@ def __init__(self, config):
         super().__init__(config)
         self.model = GlmModel(config)
         self.post_init()
+
+
+__all__ = [
+    "GlmConfig",
+    "GlmModel",
+    "GlmForCausalLM",
+    "GlmForSequenceClassification",
+    "GlmForTokenClassification",
+]