zhiweili
commited on
Commit
Β·
7966432
1
Parent(s):
6c910d6
test CUDA_HOME
Browse files- 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"
|