Skip to content

Commit 2333917

Browse files
committed
fix naming override
Signed-off-by: HuiyingLi <[email protected]>
1 parent 1939fc4 commit 2333917

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nemo_automodel/_transformers/registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,9 @@ class _ModelRegistry:
3737
naming_override: Dict[str, str] = field(default_factory=dict)
3838

3939
def __post_init__(self):
40+
self.naming_override["Qwen3OmniMoeThinkerForConditionalGeneration"] = "Qwen3OmniMoeForConditionalGeneration"
4041
for modeling_path in self.modeling_path:
4142
self._mapping_model_arch_name_to_cls(modeling_path)
42-
self.naming_override["Qwen3OmniMoeThinkerForConditionalGeneration"] = "Qwen3OmniMoeForConditionalGeneration"
4343

4444
@property
4545
def supported_models(self) -> Dict[str, Type[nn.Module]]:

0 commit comments

Comments
 (0)