fix bug of onediff.infer_compiler.register (#885)

siliconflow · May 17, 2024 · 4e4d464 · 4e4d464
1 parent d8a6a90
commit 4e4d464
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/onediff_sd_webui_extensions/compile_ldm.py b/onediff_sd_webui_extensions/compile_ldm.py
@@ -1,7 +1,7 @@
 import os
 import oneflow as flow
-from onediff.infer_compiler import oneflow_compile, register
-from onediff.infer_compiler.transform import proxy_class
+from onediff.infer_compiler import oneflow_compile
+from onediff.infer_compiler.transform import proxy_class, register
 
 from ldm.modules.attention import (
     BasicTransformerBlock,

diff --git a/onediff_sd_webui_extensions/compile_sgm.py b/onediff_sd_webui_extensions/compile_sgm.py
@@ -1,6 +1,6 @@
 import oneflow as flow
-from onediff.infer_compiler import oneflow_compile, register
-from onediff.infer_compiler.transform import proxy_class
+from onediff.infer_compiler import oneflow_compile
+from onediff.infer_compiler.transform import proxy_class, register
 from sd_webui_onediff_utils import (
     CrossAttentionOflow,
     GroupNorm32Oflow,

diff --git a/onediff_sd_webui_extensions/compile_vae.py b/onediff_sd_webui_extensions/compile_vae.py
@@ -1,8 +1,8 @@
 from modules import shared
 from modules.sd_vae_approx import model as get_vae_model, sd_vae_approx_models
 from modules.sd_vae_approx import VAEApprox
-from onediff.infer_compiler import oneflow_compile, register
-from onediff.infer_compiler.transform import proxy_class
+from onediff.infer_compiler import oneflow_compile
+from onediff.infer_compiler.transform import proxy_class, register
 
 __all__ = ["VaeCompileCtx"]