add cuda toolkit

app_onediff.py

@@ -47,7 +47,7 @@ print(os.environ.get('CUDA_PATH'))
 print(os.environ.get('CUDA_HOME'))
 os.system("pip show torch")
 
-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

requirements.txt

@@ -9,5 +9,7 @@ git+https://github.com/XPixelGroup/BasicSR@master
 gfpgan
 facexlib
 realesrgan
+cuda-toolkit
+sfast
 triton
 xformers