riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 10:12:23 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Add code sample docstrings and checkpoint reference for GLM models (#34360)

Browse Source 
* Add code sample docstrings and checkpoint reference for GLM models

* Update modular_glm.py

* Update modeling_glm.py
This commit is contained in:
Vijay 2024-10-24 20:58:51 +05:30 committed by GitHub
parent 6432ad8bb5
commit 450b9cbfac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/glm/modeling_glm.py
View File

@ -59,6 +59,9 @@ from ...processing_utils import Unpack
_CHECKPOINT_FOR_DOC = "dummy"
_CHECKPOINT_FOR_DOC = "THUDM/glm-4-9b"
class GlmRMSNorm(nn.Module):
def __init__(self, hidden_size, eps=1e-6):
"""

2
src/transformers/models/glm/modular_glm.py
View File

@ -44,6 +44,8 @@ from ..phi3.modeling_phi3 import (
from .configuration_glm import GlmConfig
_CHECKPOINT_FOR_DOC = "THUDM/glm-4-9b"
logger = logging.get_logger(__name__)
_CHECKPOINT_FOR_DOC = "dummy"