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"
|