use typing Dict

Author: DavidBert

src/diffusers/models/transformers/transformer_photon.py

@@ -427,7 +427,7 @@ def forward(
         temb: torch.Tensor,
         image_rotary_emb: torch.Tensor,
         attention_mask: Optional[torch.Tensor] = None,
-        **kwargs: dict[str, Any],
+        **kwargs: Dict[str, Any],
     ) -> torch.Tensor:
         r"""
         Runs modulation-gated cross-attention and MLP, with residual connections.