test CUDA_HOME

app_onediff.py

@@ -47,6 +47,11 @@ def runcmd(cmd, verbose = False, *args, **kwargs):
 #     options = json.loads('{"mode": "max-optimize:max-autotune:low-precision", "memory_format": "channels_last", "dynamic": true}')
 #     return compile_pipe(torch_module, backend="nexfort", options=options, fuse_qkv_projections=True)
 
+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")  
+
 BASE_MODEL = "stabilityai/sdxl-turbo"
 device = "cuda"
 

requirements.txt

@@ -9,6 +9,5 @@ git+https://github.com/XPixelGroup/BasicSR@master
 gfpgan
 facexlib
 realesrgan
-git+https://github.com/chengzeyi/stable-fast.git@main#egg=stable-fast
 triton
 xformers