Add support for training with an alpha channel for Qwen Image 2.1

This commit is contained in:
Jaret Burkett committed 2026-09-20 14:49:53 -06:00
1 parent 31ddc709c3
commit 086b663bba
8 files changed
+114 -17

No files matched your search

+4
View File
@@ -1362,6 +1362,10 @@ class GenerateImageConfig:
if self.output_ext == 'mp3':
add_album_artwork(audio_path)
else:
if image.mode == 'RGBA' and self.output_ext not in ['png', 'webp']:
# jpg cannot carry alpha, and dropping it silently would hide
# the transparency an RGBA model just generated
self.output_ext = 'png'
# TODO save image gen header info for A1111 and us, our seeds probably wont match
image.save(self.get_image_path(count, max_count))
# do prompt file