[NemotronH] Fix weight-loading unit test broken by Puzzle support (#26522)
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.7
parent
0abe6a85a5
commit
68e5b4fdd6
@@ -35,7 +35,7 @@ class _FakeParam:
|
||||
class TestNemotronHWeightLoading(unittest.TestCase):
|
||||
def _make_minimal_model(self, named_parameters=()):
|
||||
model = object.__new__(NemotronHForCausalLM)
|
||||
model.config = SimpleNamespace(n_routed_experts=2)
|
||||
model.config = SimpleNamespace(n_routed_experts=2, max_n_routed_experts=2)
|
||||
model.model = SimpleNamespace()
|
||||
model.pp_group = _FakePPGroup()
|
||||
model.remap_prefix = {}
|
||||
|
||||
Reference in New Issue
Block a user