zhiweili commited on
Commit
7966432
Β·
1 Parent(s): 6c910d6

test CUDA_HOME

Browse files
Files changed (1) hide show
  1. app_onediff.py +5 -5
app_onediff.py CHANGED
@@ -12,10 +12,6 @@ from diffusers import (
12
  AutoencoderKL,
13
  )
14
 
15
- import xformers
16
- import triton
17
- from sfast.compilers.diffusion_pipeline_compiler import (compile, CompilationConfig)
18
-
19
  def runcmd(cmd, verbose = False, *args, **kwargs):
20
 
21
  process = subprocess.Popen(
@@ -50,7 +46,11 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
50
  print(os.environ.get('CUDA_PATH'))
51
  print(os.environ.get('CUDA_HOME'))
52
 
53
- os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
 
 
 
 
54
 
55
  BASE_MODEL = "stabilityai/sdxl-turbo"
56
  device = "cuda"
 
12
  AutoencoderKL,
13
  )
14
 
 
 
 
 
15
  def runcmd(cmd, verbose = False, *args, **kwargs):
16
 
17
  process = subprocess.Popen(
 
46
  print(os.environ.get('CUDA_PATH'))
47
  print(os.environ.get('CUDA_HOME'))
48
 
49
+ os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")
50
+
51
+ import xformers
52
+ import triton
53
+ from sfast.compilers.diffusion_pipeline_compiler import (compile, CompilationConfig)
54
 
55
  BASE_MODEL = "stabilityai/sdxl-turbo"
56
  device = "cuda"