Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix OmniGen

Browse Source
This commit is contained in:
Disty0
2025-06-14 19:55:43 +03:00
parent fd583523f7
commit 24194201cf
4 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+2 -2
View File
@@ -256,8 +256,8 @@ def vae_postprocess(tensor, model, output_type='np'):
if output_type == "pil":
images = model.numpy_to_pil(images)
else:
import diffusers
model.image_processor = diffusers.image_processor.VaeImageProcessor()
from diffusers.image_processor import VaeImageProcessor
model.image_processor = VaeImageProcessor()
images = model.image_processor.postprocess(tensor, output_type=output_type)
else:
images = tensor if isinstance(tensor, list) or isinstance(tensor, np.ndarray) else [tensor]