Skip to content

Commit

Permalink
fix(api): remove device and patch calls to fix SD upscaling
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed May 9, 2023
1 parent 00e3fe5 commit eb100f6
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions api/onnx_web/diffusers/load.py
Original file line number Diff line number Diff line change
Expand Up @@ -340,28 +340,26 @@ def load_pipeline(
pipe.set_progress_bar_config(disable=True)

optimize_pipeline(server, pipe)

# TODO: remove this, not relevant with ONNX
if device is not None and hasattr(pipe, "to"):
pipe = pipe.to(device.torch_str())

# monkey-patch pipeline
patch_pipeline(server, pipe, pipeline_class, params)

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

pipe.vae_encoder.set_tiled(tiled=params.tiled_vae)
pipe.vae_decoder.set_tiled(tiled=params.tiled_vae)
if hasattr(pipe, "vae_decoder"):
pipe.vae_decoder.set_tiled(tiled=params.tiled_vae)
if hasattr(pipe, "vae_encoder"):
pipe.vae_encoder.set_tiled(tiled=params.tiled_vae)

# update panorama params
if pipeline == "panorama":
latent_window = params.tiles // 8
latent_stride = params.stride // 8

pipe.set_window_size(latent_window, latent_stride)
pipe.vae_encoder.set_window_size(latent_window, params.overlap)
pipe.vae_decoder.set_window_size(latent_window, params.overlap)
if hasattr(pipe, "vae_decoder"):
pipe.vae_decoder.set_window_size(latent_window, params.overlap)
if hasattr(pipe, "vae_encoder"):
pipe.vae_encoder.set_window_size(latent_window, params.overlap)

return pipe

Expand Down

0 comments on commit eb100f6

Please sign in to comment.