Skip to content

Commit b4408ae

Browse files
committed
Temporarily disable standalone transfer workflow
1 parent 5191d32 commit b4408ae

1 file changed

Lines changed: 10 additions & 3 deletions

File tree

src/diffusers/modular_pipelines/cosmos/modular_blocks_cosmos3.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -903,8 +903,6 @@ class Cosmos3OmniBlocks(SequentialPipelineBlocks):
903903
- `action_policy`: requires `prompt`, `action`
904904
- `action_forward_dynamics`: requires `prompt`, `action`
905905
- `action_inverse_dynamics`: requires `prompt`, `action`
906-
- `transfer`: requires `prompt`, `control_videos`
907-
908906
Components:
909907
video_processor (`VideoProcessor`)
910908
text_tokenizer (`AutoTokenizer`)
@@ -1018,13 +1016,22 @@ class Cosmos3OmniBlocks(SequentialPipelineBlocks):
10181016
"action_policy": {"prompt": True, "action": True},
10191017
"action_forward_dynamics": {"prompt": True, "action": True},
10201018
"action_inverse_dynamics": {"prompt": True, "action": True},
1021-
"transfer": {"prompt": True, "control_videos": True},
10221019
}
10231020

10241021
@property
10251022
def description(self):
10261023
return "Modular pipeline blocks for Cosmos3 generation modes."
10271024

1025+
def get_workflow(self, workflow_name: str):
1026+
if workflow_name == "transfer":
1027+
raise NotImplementedError(
1028+
'The standalone "transfer" workflow is temporarily unavailable because its nested autoregressive '
1029+
"chunk and denoising loops cannot be preserved by the current workflow extraction logic. Transfer "
1030+
"remains available through the full Cosmos3OmniBlocks pipeline. The standalone workflow will be "
1031+
"enabled after migration to the upcoming composable nested-loop abstraction."
1032+
)
1033+
return super().get_workflow(workflow_name)
1034+
10281035
@property
10291036
def outputs(self):
10301037
return [

0 commit comments

Comments
 (0)