From d636ce3eefb3033ff1aada1a24116b765403a2fe Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 17:21:12 -0600 Subject: [PATCH] fix(api): avoid setting device on schedulers that do not support it --- api/onnx_web/diffusion/load.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py index 36dfb643..a070f4e3 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -68,7 +68,7 @@ def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, schedu scheduler=scheduler.from_pretrained(model, subfolder='scheduler') ) - if device is not None: + if device is not None and hasattr(pipe, 'to'): pipe = pipe.to(device) last_pipeline_instance = pipe @@ -80,7 +80,7 @@ def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, schedu scheduler = scheduler.from_pretrained( model, subfolder='scheduler') - if device is not None: + if device is not None and hasattr(scheduler, 'to'): scheduler = scheduler.to(device) pipe.scheduler = scheduler