use nexfort_compile

app_onediff.py

@@ -14,7 +14,10 @@ os.system("python3 -m pip --no-cache-dir install --pre nexfort -f https://github
 os.system("python3 -m pip install --pre onediff")
 
 import nexfort as flow
-from onediff.infer_compiler import oneflow_compile
+from onediff.compile import onediff_compile
+
+def nexfort_compile(torch_module: torch.nn.Module, *, options=None) -> onediff.DeployableModule:
+    return onediff_compile(torch_module, backend="nexfort", options=options)
 
 BASE_MODEL = "stabilityai/sdxl-turbo"
 device = "cuda"
@@ -39,7 +42,7 @@ base_pipe.scheduler = DDPMScheduler.from_pretrained(
     subfolder="scheduler",
 )
 
-base_pipe.unet = oneflow_compile(base_pipe.unet)
+base_pipe.unet = nexfort_compile(base_pipe.unet)
 
 def create_demo() -> gr.Blocks: