zhiweili
commited on
Commit
·
347ed1e
1
Parent(s):
732c56a
test import
Browse files- app_onediff.py +1 -0
app_onediff.py
CHANGED
|
@@ -16,6 +16,7 @@ os.system("git clone https://github.com/siliconflow/onediff.git")
|
|
| 16 |
os.system("cd onediff && python3 -m pip install -e .")
|
| 17 |
os.system("cd onediff/onediff_diffusers_extensions && python3 -m pip install -e .")
|
| 18 |
|
|
|
|
| 19 |
from onediffx import compile_pipe
|
| 20 |
|
| 21 |
def nexfort_compile(torch_module: torch.nn.Module):
|
|
|
|
| 16 |
os.system("cd onediff && python3 -m pip install -e .")
|
| 17 |
os.system("cd onediff/onediff_diffusers_extensions && python3 -m pip install -e .")
|
| 18 |
|
| 19 |
+
from onediff.infer_compiler import oneflow_compile
|
| 20 |
from onediffx import compile_pipe
|
| 21 |
|
| 22 |
def nexfort_compile(torch_module: torch.nn.Module):
|