fix(api): update panorama params when using cached pipeline

ssube · May 2, 2023 · 55a1e46 · 55a1e46
1 parent a44b534
commit 55a1e46
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -125,6 +125,12 @@ def load_pipeline(
         logger.debug("reusing existing diffusion pipeline")
         pipe = cache_pipe
 
+        # update panorama params
+        if pipeline == "panorama":
+            cache_pipe.window = params.tiles // 8
+            cache_pipe.stride = params.stride() // 8
+
+        # update scheduler
         cache_scheduler = server.cache.get("scheduler", scheduler_key)
         if cache_scheduler is None:
             logger.debug("loading new diffusion scheduler")