Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

full codespell coverage

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-04 12:36:10 +02:00
parent 0bfcdabbbb
commit 5e99dee3c2
103 changed files with 264 additions and 254 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/seedvr/src/common/diffusion/samplers/base.py
+1 -1
View File
@@ -63,7 +63,7 @@ class Sampler(ABC):
f: Callable[[SamplerModelArgs], torch.Tensor],
) -> torch.Tensor:
"""
Generate a new sample given the the intial sample x and score function f.
Generate a new sample given the the initial sample x and score function f.
"""
def get_next_timestep(
modules/seedvr/src/common/distributed/ops.py
+2 -2
View File
@@ -193,10 +193,10 @@ def gather_seq_scatter_heads_qkv(
restore_shape: bool = True,
):
"""
A func to sync splited qkv tensor
A func to sync split qkv tensor
qkv_tensor: the tensor we want to do alltoall with. The last dim must
be the projection_idx, which we will split into 3 part. After
spliting, the gather idx will be projecttion_idx + 1
splitting, the gather idx will be projecttion_idx + 1
seq_dim: gather_dim for all2all comm
restore_shape: if True, output will has the same shape length as input
"""