zhiweili
commited on
Commit
Β·
cfe1559
1
Parent(s):
8303ab4
try to fix path
Browse files- app_onediff.py +7 -2
app_onediff.py
CHANGED
@@ -13,10 +13,15 @@ from diffusers import (
|
|
13 |
|
14 |
os.system("python3 -m pip --no-cache-dir install --pre nexfort -f https://github.com/siliconflow/nexfort_releases/releases/expanded_assets/torch2.4.1_cu121")
|
15 |
os.system("git clone https://github.com/siliconflow/onediff.git")
|
16 |
-
os.system("python3 -m pip install -e
|
17 |
-
os.system("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):
|
|
|
13 |
|
14 |
os.system("python3 -m pip --no-cache-dir install --pre nexfort -f https://github.com/siliconflow/nexfort_releases/releases/expanded_assets/torch2.4.1_cu121")
|
15 |
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 |
+
import sys
|
20 |
+
print(sys.version)
|
21 |
+
|
22 |
+
sys.path.append("onediff")
|
23 |
from onediff.infer_compiler import oneflow_compile
|
24 |
+
sys.path.append("onediff/onediff_diffusers_extensions")
|
25 |
from onediffx import compile_pipe
|
26 |
|
27 |
def nexfort_compile(torch_module: torch.nn.Module):
|