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