ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small tweaks and bug fixes and future proofing

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-05 12:39:45 -06:00
parent ead23cee88
commit 2b901cca39
4 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions_built_in/flex2/flex2.py
View File

@@ -424,7 +424,9 @@ class Flex2(BaseModel):
if self.random_blur_mask:
# blur the mask
# Give it a channel dim of 1
inpainting_tensor_mask = inpainting_tensor_mask.unsqueeze(1)
if len(inpainting_tensor_mask.shape) == 3:
# if it is 3d, add a channel dim
inpainting_tensor_mask = inpainting_tensor_mask.unsqueeze(1)
# we are at latent size, so keep kernel smaller
inpainting_tensor_mask = random_blur(
inpainting_tensor_mask,
@@ -432,8 +434,6 @@ class Flex2(BaseModel):
max_kernel_size=8,
p=0.5
)
# remove the channel dim
inpainting_tensor_mask = inpainting_tensor_mask.squeeze(1)
do_mask_invert = False
if self.invert_inpaint_mask_chance > 0.0: