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

update mask blur

Browse Source
This commit is contained in:
Artiprocher
2024-11-12 19:20:17 +08:00
parent d46b8b8fd7
commit 71b17a3a53
1 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
diffsynth/pipelines/base.py
View File

@@ -36,7 +36,8 @@ class BasePipeline(torch.nn.Module):
return video return video
def merge_latents(self, value, latents, masks, scales, blur_kernel_size=3, blur_sigma=1.0): def merge_latents(self, value, latents, masks, scales, blur_kernel_size=33, blur_sigma=10.0):
if len(latents) > 0:
blur = GaussianBlur(kernel_size=blur_kernel_size, sigma=blur_sigma) blur = GaussianBlur(kernel_size=blur_kernel_size, sigma=blur_sigma)
height, width = value.shape[-2:] height, width = value.shape[-2:]
weight = torch.ones_like(value) weight = torch.ones_like(value)