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

swtich to igwn-segments

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-16 18:15:59 +02:00
parent 5b0a732db3
commit ae8d559b50
5 changed files with 22 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/mod/__init__.py
+5 -3
View File
@@ -52,9 +52,9 @@ from diffusers.utils.torch_utils import randn_tensor
try:
from ligo.segments import segment
from igwn_segments import segment
except ImportError as e:
raise ImportError("Please install transformers and ligo-segments to use the mixture pipeline") from e
raise ImportError("Please install transformers and igwn-segments to use the mixture pipeline") from e
if is_torch_xla_available():
import torch_xla.core.xla_model as xm
@@ -1138,7 +1138,9 @@ class StableDiffusionXLTilingPipeline(
else guidance_scale_tiles[row][col]
)
noise_pred_tile = noise_pred_uncond + guidance * (noise_pred_text - noise_pred_uncond)
noise_preds_row.append(noise_pred_tile)
else:
noise_pred_tile = noise_pred
noise_preds_row.append(noise_pred_tile)
noise_preds.append(noise_preds_row)
# Stitch noise predictions for all tiles