theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
Artiprocher
2025-08-04 20:35:44 +08:00
parent c35f2d8bda
commit 1625894694
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/pipelines/qwen_image.py
View File

@@ -235,7 +235,7 @@ class QwenImagePipeline(BasePipeline):
inputs_shared["latents"] = self.scheduler.step(noise_pred, self.scheduler.timesteps[progress_id], inputs_shared["latents"]) inputs_shared["latents"] = self.scheduler.step(noise_pred, self.scheduler.timesteps[progress_id], inputs_shared["latents"])
# Decode # Decode
self.load_models_to_device(['vae_decoder']) self.load_models_to_device(['vae'])
image = self.vae.decode(inputs_shared["latents"], device=self.device, tiled=tiled, tile_size=tile_size, tile_stride=tile_stride) image = self.vae.decode(inputs_shared["latents"], device=self.device, tiled=tiled, tile_size=tile_size, tile_stride=tile_stride)
image = self.vae_output_to_image(image) image = self.vae_output_to_image(image)
self.load_models_to_device([]) self.load_models_to_device([])
@@ -268,7 +268,7 @@ class QwenImageUnit_InputImageEmbedder(PipelineUnit):
def __init__(self): def __init__(self):
super().__init__( super().__init__(
input_params=("input_image", "noise", "tiled", "tile_size", "tile_stride"), input_params=("input_image", "noise", "tiled", "tile_size", "tile_stride"),
onload_model_names=("vae_encoder",) onload_model_names=("vae",)
) )
def process(self, pipe: QwenImagePipeline, input_image, noise, tiled, tile_size, tile_stride): def process(self, pipe: QwenImagePipeline, input_image, noise, tiled, tile_size, tile_stride):