nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Import Tuple from typing instead of torch

Browse source
This commit is contained in:
Alex Gherghina 2025-03-25 10:38:55 +02:00
parent 4c3c1740a3
commit 8202ca13df
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
opensora/models/mmdit/math.py
View file

@ -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