fix(api): leave pipelines on default device unless specified

ssube · Jan 21, 2023 · 505cacf · 505cacf
1 parent 52484e6
commit 505cacf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/pipeline.py b/api/onnx_web/pipeline.py
@@ -44,7 +44,7 @@ def get_latents_from_seed(seed: int, size: Size) -> np.ndarray:
     return image_latents
 
 
-def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, scheduler: Any, device: Union[str, None]):
+def load_pipeline(pipeline: DiffusionPipeline, model: str, provider: str, scheduler: Any, device: Union[str, None] = None):
     global last_pipeline_instance
     global last_pipeline_scheduler
     global last_pipeline_options