zhiweili
commited on
Commit
Β·
83b6a09
1
Parent(s):
54bc1d3
add oneflow
Browse files- app_onediff.py +3 -2
app_onediff.py
CHANGED
@@ -10,10 +10,11 @@ from diffusers import (
|
|
10 |
AutoencoderTiny,
|
11 |
)
|
12 |
|
13 |
-
|
14 |
os.system("pip install --pre onediff")
|
|
|
15 |
|
16 |
-
|
17 |
from onediff.infer_compiler import oneflow_compile
|
18 |
|
19 |
BASE_MODEL = "stabilityai/sdxl-turbo"
|
|
|
10 |
AutoencoderTiny,
|
11 |
)
|
12 |
|
13 |
+
|
14 |
os.system("pip install --pre onediff")
|
15 |
+
os.system("pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/community_cu122")
|
16 |
|
17 |
+
import oneflow as flow
|
18 |
from onediff.infer_compiler import oneflow_compile
|
19 |
|
20 |
BASE_MODEL = "stabilityai/sdxl-turbo"
|