VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix NaN in sampling image

Browse Source
This commit is contained in:
Kohya S
2023-07-11 23:18:35 +09:00
parent 2e67d74df4
commit 814996b14f
2 changed files with 118 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sdxl_lpw_stable_diffusion.py
View File

@@ -922,7 +922,7 @@ class SdxlStableDiffusionLongPromptWeightingPipeline:
if up1 is not None: if up1 is not None:
uncond_pool = up1 uncond_pool = up1
dtype = text_embeddings_list[0].dtype dtype = self.unet.dtype
# 4. Preprocess image and mask # 4. Preprocess image and mask
if isinstance(image, PIL.Image.Image): if isinstance(image, PIL.Image.Image):

2
library/train_util.py
View File

@@ -3874,7 +3874,7 @@ def sample_images_common(
cuda_rng_state = torch.cuda.get_rng_state() if torch.cuda.is_available() else None cuda_rng_state = torch.cuda.get_rng_state() if torch.cuda.is_available() else None
with torch.no_grad(): with torch.no_grad():
with accelerator.autocast(): # with accelerator.autocast():
for i, prompt in enumerate(prompts): for i, prompt in enumerate(prompts):
if not accelerator.is_main_process: if not accelerator.is_main_process:
continue continue