Skip to content

Commit aed1f19

Browse files
DavidBertdg845
andcommitted
Use Tuple instead of tuple
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
1 parent 1354f45 commit aed1f19

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/diffusers/models/transformers/transformer_photon.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def __init__(self, dim: int):
340340
nn.init.constant_(self.lin.weight, 0)
341341
nn.init.constant_(self.lin.bias, 0)
342342

343-
def forward(self, vec: Tensor) -> tuple[tuple[Tensor, Tensor, Tensor], tuple[Tensor, Tensor, Tensor]]:
343+
def forward(self, vec: Tensor) -> Tuple[Tuple[Tensor, Tensor, Tensor], Tuple[Tensor, Tensor, Tensor]]:
344344
out = self.lin(nn.functional.silu(vec))[:, None, :].chunk(6, dim=-1)
345345
return tuple(out[:3]), tuple(out[3:])
346346

0 commit comments

Comments
 (0)