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

Merge pull request #774 from lansing/lansing/sdxl-fix-gen-memleak

Browse Source 
fix: VRAM memory leak in sdxl_gen_img.py
This commit is contained in:
Kohya S
2023-09-02 15:20:32 +09:00
committed by GitHub
parent f77bdf96d8 fbc550d02e
commit 6400116715
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sdxl_gen_img.py Normal file → Executable file
View File

@@ -761,6 +761,9 @@ class PipelineLike:
# we always cast to float32 as this does not cause significant overhead and is compatible with bfloa16 # we always cast to float32 as this does not cause significant overhead and is compatible with bfloa16
image = image.cpu().permute(0, 2, 3, 1).float().numpy() image = image.cpu().permute(0, 2, 3, 1).float().numpy()
if torch.cuda.is_available():
torch.cuda.empty_cache()
if output_type == "pil": if output_type == "pil":
# image = self.numpy_to_pil(image) # image = self.numpy_to_pil(image)
image = (image * 255).round().astype("uint8") image = (image * 255).round().astype("uint8")