Skip to content

Commit eb100f6

Browse files
committed
fix(api): remove device and patch calls to fix SD upscaling
1 parent 00e3fe5 commit eb100f6

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

api/onnx_web/diffusers/load.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -340,28 +340,26 @@ def load_pipeline(
340340
pipe.set_progress_bar_config(disable=True)
341341

342342
optimize_pipeline(server, pipe)
343-
344-
# TODO: remove this, not relevant with ONNX
345-
if device is not None and hasattr(pipe, "to"):
346-
pipe = pipe.to(device.torch_str())
347-
348-
# monkey-patch pipeline
349343
patch_pipeline(server, pipe, pipeline_class, params)
350344

351345
server.cache.set("diffusion", pipe_key, pipe)
352346
server.cache.set("scheduler", scheduler_key, components["scheduler"])
353347

354-
pipe.vae_encoder.set_tiled(tiled=params.tiled_vae)
355-
pipe.vae_decoder.set_tiled(tiled=params.tiled_vae)
348+
if hasattr(pipe, "vae_decoder"):
349+
pipe.vae_decoder.set_tiled(tiled=params.tiled_vae)
350+
if hasattr(pipe, "vae_encoder"):
351+
pipe.vae_encoder.set_tiled(tiled=params.tiled_vae)
356352

357353
# update panorama params
358354
if pipeline == "panorama":
359355
latent_window = params.tiles // 8
360356
latent_stride = params.stride // 8
361357

362358
pipe.set_window_size(latent_window, latent_stride)
363-
pipe.vae_encoder.set_window_size(latent_window, params.overlap)
364-
pipe.vae_decoder.set_window_size(latent_window, params.overlap)
359+
if hasattr(pipe, "vae_decoder"):
360+
pipe.vae_decoder.set_window_size(latent_window, params.overlap)
361+
if hasattr(pipe, "vae_encoder"):
362+
pipe.vae_encoder.set_window_size(latent_window, params.overlap)
365363

366364
return pipe
367365

0 commit comments

Comments
 (0)