Skip to content

Commit

Permalink
Close Zamba2Config code block (huggingface#35914)
Browse files Browse the repository at this point in the history
* close zamba2 code block

* Add Zamba2 to toctree
  • Loading branch information
Rocketknight1 authored and elvircrn committed Feb 13, 2025
1 parent 246c193 commit a26337f
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
2 changes: 2 additions & 0 deletions docs/source/en/_toctree.yml
Original file line number Diff line number Diff line change
Expand Up @@ -626,6 +626,8 @@
title: YOSO
- local: model_doc/zamba
title: Zamba
- local: model_doc/zamba2
title: Zamba2
title: Text models
- isExpanded: false
sections:
Expand Down
2 changes: 1 addition & 1 deletion src/transformers/models/zamba2/configuration_zamba2.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]
Expand Down

0 comments on commit a26337f

Please sign in to comment.