Add code sample docstrings and checkpoint reference for GLM models (h…

…uggingface#34360) * Add code sample docstrings and checkpoint reference for GLM models * Update modular_glm.py * Update modeling_glm.py
innovationcore · Dec 6, 2024 · 98e5571 · 98e5571
1 parent f5ab88d
commit 98e5571
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/transformers/models/glm/modeling_glm.py b/src/transformers/models/glm/modeling_glm.py
@@ -59,6 +59,9 @@
 _CHECKPOINT_FOR_DOC = "dummy"
 
 
+_CHECKPOINT_FOR_DOC = "THUDM/glm-4-9b"
+
+
 class GlmRMSNorm(nn.Module):
     def __init__(self, hidden_size, eps=1e-6):
         """

diff --git a/src/transformers/models/glm/modular_glm.py b/src/transformers/models/glm/modular_glm.py
@@ -44,6 +44,8 @@
 from .configuration_glm import GlmConfig
 
 
+_CHECKPOINT_FOR_DOC = "THUDM/glm-4-9b"
+
 logger = logging.get_logger(__name__)
 
 _CHECKPOINT_FOR_DOC = "dummy"