-
Notifications
You must be signed in to change notification settings - Fork 328
Open
Labels
GemmaGemma model specific issuesGemma model specific issues
Description
Description
I noticed several inconsistencies in the documentation within the keras_hub/src/models/ directory. These include arguments present in __init__ methods but missing from the corresponding class docstrings, errors in code examples, and various typographical errors spread across multiple model files.
Summary of issues identified
Missing arguments in docstrings
- MistralBackbone: The
dropoutargument is missing from theArgssection. - FluxBackbone: Five arguments are missing from the
Argssection:image_shapetext_shapeimage_ids_shapetext_ids_shapey_shape
Code examples
- PaliGemmaBackbone: The usage example contains a typo (
images_sizeinstead ofimage_size), which causes the example to fail if executed.
Typos & grammar
roataryinstead ofrotary(GPT-OSS, Llama3, Mixtral).traingularinstead oftriangular(Mixtral).Pollinginstead ofPoolingin error messages (HGNetV2, VGG, ImageClassifier).done a float32instead ofdone in float32(PaliGemma, CLIP, SigLIP, DinoV2/DinoV3, etc.).value user forinstead ofvalue used for(Gemma, Gemma3).fucntioninstead offunction.
I have fixed these docstrings locally to match the actual code definitions and prepared a pull request to address all these instances.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
GemmaGemma model specific issuesGemma model specific issues