SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Close Zamba2Config code block (#35914)

Browse Source 
* close zamba2 code block

* Add Zamba2 to toctree
This commit is contained in:
Matt
2025-01-27 19:09:42 +00:00
committed by GitHub
parent 63e9c941eb
commit 414658f94f
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/en/_toctree.yml
View File

@@ -626,6 +626,8 @@
title: YOSO title: YOSO
- local: model_doc/zamba - local: model_doc/zamba
title: Zamba title: Zamba
- local: model_doc/zamba2
title: Zamba2
title: Text models title: Text models
- isExpanded: false - isExpanded: false
sections: sections:

2
src/transformers/models/zamba2/configuration_zamba2.py
View File

@@ -120,7 +120,7 @@ class Zamba2Config(PretrainedConfig):
>>> model = Zamba2Model(configuration) >>> model = Zamba2Model(configuration)
>>> # Accessing the model configuration >>> # Accessing the model configuration
>>> configuration = model.config >>> configuration = model.config
""" ```"""
model_type = "zamba2" model_type = "zamba2"
keys_to_ignore_at_inference = ["past_key_values"] keys_to_ignore_at_inference = ["past_key_values"]