@@ -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