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

Merge pull request #839 from alexandru-g/fix/math-import

Browse source 
Import Tuple from typing instead of torch
This commit is contained in:
Xiwen Wu 2025-03-25 17:29:43 +08:00 committed by GitHub
commit 3455b7e0fb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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