zhiweili
commited on
Commit
Β·
c51baca
1
Parent(s):
8f95526
fix import
Browse files- 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.
|
|
|
|
|
|
|
18 |
|
19 |
-
def nexfort_compile(torch_module: torch.nn.Module, *, options=None) ->
|
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"
|