zhiweili commited on
Commit
5018a4b
Β·
1 Parent(s): 6981f48

add onediff

Browse files
Files changed (2) hide show
  1. app_onediff.py +0 -4
  2. requirements.txt +4 -2
app_onediff.py CHANGED
@@ -10,10 +10,6 @@ from diffusers import (
10
  AutoencoderTiny,
11
  )
12
 
13
-
14
- os.system("pip install --pre oneflow")
15
- os.system("pip install --pre onediff")
16
-
17
  import oneflow as flow
18
  from onediff.infer_compiler import oneflow_compile
19
 
 
10
  AutoencoderTiny,
11
  )
12
 
 
 
 
 
13
  import oneflow as flow
14
  from onediff.infer_compiler import oneflow_compile
15
 
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
  gradio
2
- torch
3
- torchvision
4
  diffusers
5
  transformers
6
  accelerate
7
  spaces
 
 
 
1
  gradio
2
+ torch -f https://download.pytorch.org/whl/cu121/torch_stable.html
3
+ torchvision -f https://download.pytorch.org/whl/cu121/torch_stable.html
4
  diffusers
5
  transformers
6
  accelerate
7
  spaces
8
+ oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/community_cu121
9
+ onediff