zhiweili commited on
Commit
c51baca
Β·
1 Parent(s): 8f95526

fix import

Browse files
Files changed (1) hide show
  1. app_onediff.py +5 -2
app_onediff.py CHANGED
@@ -14,9 +14,12 @@ os.system("python3 -m pip --no-cache-dir install --pre nexfort -f https://github
14
  os.system("python3 -m pip install --pre onediff")
15
 
16
  import nexfort as flow
17
- from onediff.compile import onediff_compile
 
 
 
18
 
19
- def nexfort_compile(torch_module: torch.nn.Module, *, options=None) -> onediff.DeployableModule:
20
  return onediff_compile(torch_module, backend="nexfort", options=options)
21
 
22
  BASE_MODEL = "stabilityai/sdxl-turbo"
 
14
  os.system("python3 -m pip install --pre onediff")
15
 
16
  import nexfort as flow
17
+ from onediff.infer_compiler import (
18
+ compile as onediff_compile,
19
+ DeployableModule,
20
+ )
21
 
22
+ def nexfort_compile(torch_module: torch.nn.Module, *, options=None) -> DeployableModule:
23
  return onediff_compile(torch_module, backend="nexfort", options=options)
24
 
25
  BASE_MODEL = "stabilityai/sdxl-turbo"