From 8202ca13dfd61fb6c8fe963bcf1eb193913d64a6 Mon Sep 17 00:00:00 2001 From: Alex Gherghina Date: Tue, 25 Mar 2025 10:38:55 +0200 Subject: [PATCH] Import Tuple from typing instead of torch --- opensora/models/mmdit/math.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/opensora/models/mmdit/math.py b/opensora/models/mmdit/math.py index 8257b0e..f09cd97 100644 --- a/opensora/models/mmdit/math.py +++ b/opensora/models/mmdit/math.py @@ -2,7 +2,8 @@ import torch from einops import rearrange from flash_attn import flash_attn_func as flash_attn_func_v2 from liger_kernel.ops.rope import LigerRopeFunction -from torch import Tensor, Tuple +from torch import Tensor +from typing import Tuple try: from flash_attn_interface import flash_attn_func as flash_attn_func_v3