add onediffx

app_onediff.py

@@ -10,8 +10,10 @@ from diffusers import (
     AutoencoderTiny,
 )
 
-import oneflow as flow
-from onediff.infer_compiler import oneflow_compile
+os.system("git clone https://github.com/siliconflow/onediff.git")
+os.system("cd onediff_diffusers_extensions && python3 -m pip install -e .")
+
+from onediffx import compile_pipe
 
 BASE_MODEL = "stabilityai/sdxl-turbo"
 device = "cuda"
@@ -35,8 +37,8 @@ base_pipe.scheduler = DDPMScheduler.from_pretrained(
     BASE_MODEL,
     subfolder="scheduler",
 )
-base_pipe.unet = oneflow_compile(base_pipe.unet)
-# base_pipe.vae.decoder = oneflow_compile(base_pipe.vae.decoder)
+
+base_pipe = compile_pipe(base_pipe)
 
 def create_demo() -> gr.Blocks:
 

requirements.txt

@@ -1,9 +1,7 @@
 gradio
-torch -f https://download.pytorch.org/whl/cu121/torch_stable.html
-torchvision -f https://download.pytorch.org/whl/cu121/torch_stable.html
+torch
+torchvision
 diffusers
 transformers
 accelerate
 spaces
-oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/community_cu121
-onediff