test CUDA_HOME

app_onediff.py

@@ -12,10 +12,6 @@ from diffusers import (
     AutoencoderKL,
 )
 
-import xformers
-import triton
-from sfast.compilers.diffusion_pipeline_compiler import (compile, CompilationConfig)
-
 def runcmd(cmd, verbose = False, *args, **kwargs):
 
     process = subprocess.Popen(
@@ -50,7 +46,11 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
 print(os.environ.get('CUDA_PATH'))
 print(os.environ.get('CUDA_HOME'))
 
-os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast")  
+os.system("CUDA_HOME=/usr/local/cuda-12.1 python3 -m pip install stable-fast") 
+
+import xformers
+import triton
+from sfast.compilers.diffusion_pipeline_compiler import (compile, CompilationConfig)
 
 BASE_MODEL = "stabilityai/sdxl-turbo"
 device = "cuda"