[CI/Build] Enable tests for recent day-0 new models (#34585)

Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn>
This commit is contained in:
Isotr0py
2026-02-16 10:17:04 +08:00
committed by GitHub
parent 23d825aba1
commit 91ac5d9bfd
4 changed files with 6 additions and 16 deletions

View File

@@ -85,11 +85,7 @@ class InternS1ProProcessingInfo(Qwen3VLProcessingInfo):
return self.ctx.get_hf_config()
def get_hf_processor(self, **kwargs: object) -> AutoProcessor:
return AutoProcessor.from_pretrained(
self.ctx.model_config.model,
trust_remote_code=True,
**kwargs,
)
return self.ctx.get_hf_processor(**kwargs)
class InternS1ProMoeMLP(nn.Module):
@@ -497,7 +493,7 @@ class InternS1ProMoeLLMForCausalLM(Qwen3MoeForCausalLM):
)
class Qwen3VLMoeMixtureOfExperts(MixtureOfExperts):
class InternS1ProMoeMixtureOfExperts(MixtureOfExperts):
def update_physical_experts_metadata(
self,
num_physical_experts: int,
@@ -547,7 +543,7 @@ class Qwen3VLMoeMixtureOfExperts(MixtureOfExperts):
dummy_inputs=Qwen3VLDummyInputsBuilder,
)
class InternS1ProForConditionalGeneration(
Qwen3VLForConditionalGeneration, Qwen3VLMoeMixtureOfExperts
Qwen3VLForConditionalGeneration, InternS1ProMoeMixtureOfExperts
):
is_3d_moe_weight: bool = True
packed_modules_mapping = {