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

[CI] doc builder without custom image (#36862)

Browse Source 
* no image

* test

* revert jax version updates

* make fixup

* update autodoc path for model_addition_debugger

* shieldgemma2

* add missing pages to toctree
This commit is contained in:
Joao Gante
2025-03-21 09:10:27 +00:00
committed by GitHub
parent 97d2f9d8ae
commit 949cca4061
6 changed files with 29 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/en/internal/model_debugging_utils.md
View File

@@ -26,7 +26,7 @@ Most of those are only useful if you are adding new models in the library.
### Model addition debugger - context manager for model adders
This context manager is a power user tool intended for model adders.
This context manager is a power user tool intended for model adders.
It tracks all forward calls within a model forward and logs a slice of each input and output on a nested Json.
To note, this context manager enforces `torch.inference_mode()`.
@@ -66,6 +66,6 @@ with model_addition_debugger_context(model, "optional_path_to_your_output_file.j
```
[[autodoc]] utils.model_addition_debugger
[[autodoc]] model_addition_debugger
[[autodoc]] utils.model_addition_debugger_context
[[autodoc]] model_addition_debugger_context