MMaghoumi

/

Nemotron-Labs-Diffusion-TinyStories-3b

like 0

Align the public architecture name with the canonical class used by nvidia/Nemotron-Labs-Diffusion-8B (NemotronLabsDiffusionModel). This makes both repos load in SGLang and any other downstream serving stack without per-model architectures overrides.

Changes:

• config.json: architectures and auto_map.AutoModel updated.
• modeling_nemotron_labs_diffusion.py: class + __all__ renamed.
• configuration_nemotron_labs_diffusion.py: docstring reference updated.