We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 434656b commit 924f71bCopy full SHA for 924f71b
tests/models/ernie4_5_moe/test_modeling_ernie4_5_moe.py
@@ -97,7 +97,6 @@ def test_load_balancing_loss(self):
97
config, input_dict = self.model_tester.prepare_config_and_inputs_for_common()
98
config.num_labels = 3
99
config.num_experts = 3
100
- config.expert_interval = 1
101
config.output_router_logits = True
102
input_ids = input_dict["input_ids"]
103
attention_mask = input_ids.ne(config.pad_token_id).to(torch_device)
0 commit comments