2 Commits

Author SHA1 Message Date
  Xiwen Wu 3455b7e0fb
Merge pull request #839 from alexandru-g/fix/math-import 9 months ago
  Alex Gherghina 8202ca13df Import Tuple from typing instead of torch 9 months ago
1 changed files with 2 additions and 1 deletions
Split View
  1. +2
    -1
      opensora/models/mmdit/math.py

+ 2
- 1
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


Loading…
Cancel
Save
Baidu
map