fix import

2023-01-27 17:11:27 -06:00 · 2023-01-27 17:11:27 -06:00 · bfd8e0a739
parent 71ff3bb1c4
commit bfd8e0a739
1 changed files with 3 additions and 1 deletions
--- a/api/onnx_web/diffusion.py
+++ b/api/onnx_web/diffusion.py
@ -12,6 +12,9 @@ import gc
 import numpy as np
 import torch

+from .chain import (
+    StageParams,
+)
 from .image import (
    expand_image,
 )
@ -25,7 +28,6 @@ from .utils import (
    ImageParams,
    Border,
    ServerContext,
-    StageParams,
    Size,
 )