Skip to content

Commit 9059c43

Browse files
committed
fix many type hint errors
1 parent 3107fe0 commit 9059c43

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_hunyuan_video.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1038,7 +1038,7 @@ def forward(
10381038
guidance: torch.Tensor = None,
10391039
attention_kwargs: Optional[Dict[str, Any]] = None,
10401040
return_dict: bool = True,
1041-
) -> Union[torch.Tensor, Transformer2DModelOutput]:
1041+
) -> Union[Tuple[torch.Tensor], Transformer2DModelOutput]:
10421042
if attention_kwargs is not None:
10431043
attention_kwargs = attention_kwargs.copy()
10441044
lora_scale = attention_kwargs.pop("scale", 1.0)

0 commit comments

Comments
 (0)