theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove py310 type hint

Browse Source
This commit is contained in:
Gregory D. Hunkins
2025-02-25 22:29:53 -05:00
parent bed770248b
commit 366ee0f542
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/models/stepvideo_dit.py
View File

@@ -10,7 +10,7 @@
# The above copyright notice and this permission notice shall be included in all # The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software. # copies or substantial portions of the Software.
# ============================================================================== # ==============================================================================
from typing import Dict, Optional, Tuple from typing import Dict, Optional, Tuple, Union, List
import torch, math import torch, math
from torch import nn from torch import nn
from einops import rearrange, repeat from einops import rearrange, repeat
@@ -757,7 +757,7 @@ class StepVideoModel(torch.nn.Module):
norm_elementwise_affine: bool = False, norm_elementwise_affine: bool = False,
norm_eps: float = 1e-6, norm_eps: float = 1e-6,
use_additional_conditions: Optional[bool] = False, use_additional_conditions: Optional[bool] = False,
caption_channels: Optional[int]|list|tuple = [6144, 1024], caption_channels: Optional[Union[int, List, Tuple]] = [6144, 1024],
attention_type: Optional[str] = "torch", attention_type: Optional[str] = "torch",
): ):
super().__init__() super().__init__()