If I add "unconditional_transformer" and remove "text_encoder" from the hardcoded component list in test_group_offloading_inference's enable_group_offload_on_component local function, the test fully passes locally (and if I keep "text_encoder", the block-level but not leaf-level part of the test passes). So I think we should consider
- Adding a way to configure the
test_group_offloading_inference component list by the pipeline test config class
- Splitting the leaf-level and block-level parts of the test
This also appears to contradict the skip reason message given above.
Originally posted by @dg845 in #14599 (comment)
If I add
"unconditional_transformer"and remove"text_encoder"from the hardcoded component list intest_group_offloading_inference'senable_group_offload_on_componentlocal function, the test fully passes locally (and if I keep"text_encoder", the block-level but not leaf-level part of the test passes). So I think we should considertest_group_offloading_inferencecomponent list by the pipeline test config classThis also appears to contradict the skip
reasonmessage given above.Originally posted by @dg845 in #14599 (comment)