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

fix: VRAM memory leak in sdxl_gen_img.py

Browse Source
This commit is contained in:
Max Lansing
2023-08-20 19:04:16 -07:00
parent 014c4b47c9
commit fbc550d02e
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

@@ -754,6 +754,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")