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

* 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

View File

@@ -985,6 +985,8 @@
title: Qwen2VL
- local: model_doc/sam
title: Segment Anything
- local: model_doc/shieldgemma2
title: ShieldGemma2
- local: model_doc/siglip
title: SigLIP
- local: model_doc/siglip2
@@ -1044,6 +1046,8 @@
- sections:
- local: internal/modeling_utils
title: Custom Layers and Utilities
- local: internal/model_debugging_utils
title: Utilities for Model Debugging
- local: internal/pipelines_utils
title: Utilities for pipelines
- local: internal/tokenization_utils

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