Skip to content

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 #14618

Description

@sayakpaul

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

  1. Adding a way to configure the test_group_offloading_inference component list by the pipeline test config class
  2. 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)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions