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 to work sample generation in fp8 ref #1057

Browse Source
This commit is contained in:
Kohya S
2024-01-20 10:57:42 +09:00
parent a7ef6422b6
commit 1f77bb6e73
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
library/sdxl_lpw_stable_diffusion.py
View File

@@ -923,7 +923,11 @@ class SdxlStableDiffusionLongPromptWeightingPipeline:
if up1 is not None: if up1 is not None:
uncond_pool = up1 uncond_pool = up1
dtype = self.unet.dtype unet_dtype = self.unet.dtype
dtype = unet_dtype
if dtype.itemsize == 1: # fp8
dtype = torch.float16
self.unet.to(dtype)
# 4. Preprocess image and mask # 4. Preprocess image and mask
if isinstance(image, PIL.Image.Image): if isinstance(image, PIL.Image.Image):
@@ -1028,6 +1032,7 @@ class SdxlStableDiffusionLongPromptWeightingPipeline:
if is_cancelled_callback is not None and is_cancelled_callback(): if is_cancelled_callback is not None and is_cancelled_callback():
return None return None
self.unet.to(unet_dtype)
return latents return latents
def latents_to_image(self, latents): def latents_to_image(self, latents):