From 414658f94f426a04abe7655b21c89bbf8fee99c3 Mon Sep 17 00:00:00 2001 From: Matt Date: Mon, 27 Jan 2025 19:09:42 +0000 Subject: [PATCH] Close Zamba2Config code block (#35914) * close zamba2 code block * Add Zamba2 to toctree --- docs/source/en/_toctree.yml | 2 ++ src/transformers/models/zamba2/configuration_zamba2.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/docs/source/en/_toctree.yml b/docs/source/en/_toctree.yml index 34aacd0796..1e02abd1bb 100644 --- a/docs/source/en/_toctree.yml +++ b/docs/source/en/_toctree.yml @@ -626,6 +626,8 @@ title: YOSO - local: model_doc/zamba title: Zamba + - local: model_doc/zamba2 + title: Zamba2 title: Text models - isExpanded: false sections: diff --git a/src/transformers/models/zamba2/configuration_zamba2.py b/src/transformers/models/zamba2/configuration_zamba2.py index 975e968735..fdf209d2e7 100644 --- a/src/transformers/models/zamba2/configuration_zamba2.py +++ b/src/transformers/models/zamba2/configuration_zamba2.py @@ -120,7 +120,7 @@ class Zamba2Config(PretrainedConfig): >>> model = Zamba2Model(configuration) >>> # Accessing the model configuration >>> configuration = model.config - """ + ```""" model_type = "zamba2" keys_to_ignore_at_inference = ["past_key_values"]