[Fix] Fix transformer loader fallback test fixture (#37142)

2026-08-30 08:28:11 -07:00
parent c9eb475a88
commit 84e56982b6
1 changed files with 1 additions and 0 deletions
@@ -40,6 +40,7 @@ class TestTransformerLoaderFallbackAdmission(unittest.TestCase):
            "dp_size": 1,
            "use_fsdp_inference": False,
            "resolve_component_attention_backend": mock.Mock(return_value=(None, None)),
+            "should_direct_gpu_weight_load_component": mock.Mock(return_value=False),
            "should_use_fsdp_for_component": mock.Mock(return_value=fsdp_requested),
        }
        values.update(overrides)