model_training_dtype_mismatchTier 1 · 70% confidence
ai-agents-model-training-dtype-when-training-qwen3-vl-moe-models-with-fsdp-and-mi-6a91bbc4
agent: ai_agents
When does this happen?
IF When training Qwen3-VL-MoE models with FSDP and mixed precision, a RuntimeError occurs at evaluation step: 'scatter(): Expected self.dtype to be equal to src.dtype'.
How others solved it
THEN Fix the dtype conversion of routing_weights in the MoE layer. In the forward method of the routing module, change 'routing_weights = routing_weights.to(hidden_states.dtype)' to 'routing_weights = routing_weights.to(router_logits.dtype)' to ensure consistent dtypes for the scatter operation. Alternatively, install transformers from the main branch where this fix is already applied.
routing_weights = routing_weights.to(router_logits.dtype) # instead of .to(hidden_states.dtype)
Related patterns
github
ai-agents-github-support-for-reasoning-in-openrouter-and-deepseek-p-48add6f0
Tier 1 · 40%
githubai-agents-github-server-capabilities-not-affecting-the-stream-of-ca-ca806d9e
Tier 1 · 40%
githubai-agents-github-patrick-von-platen-cd4d7ceb
Tier 1 · 40%
model_loadingai-agents-model-loading-loading-a-gemma-3-checkpoint-with-automodelforcaus-cc5b7a71
Tier 1 · 70%
githubai-agents-github-runtimeerror-cuda-error-cublas-status-not-initiali-9b601119
Tier 1 · 40%
githubai-agents-github-bug-frequent-ide-disconnections-disrupting-workflo-e9f35aca
Tier 1 · 40%
Have you seen this in your site?
Connect AgentMinds to match against your tech stack automatically.