Spaces:
Runtime error
Runtime error
Update gradio-app.py
Browse files- gradio-app.py +1 -0
gradio-app.py
CHANGED
|
@@ -45,6 +45,7 @@ if TORCH_COMPILE:
|
|
| 45 |
#pipe.vae = torch.compile(pipe.vae, mode=optmode)
|
| 46 |
doCompile = True
|
| 47 |
if doCompile:
|
|
|
|
| 48 |
config = CompilationConfig.Default()
|
| 49 |
|
| 50 |
try:
|
|
|
|
| 45 |
#pipe.vae = torch.compile(pipe.vae, mode=optmode)
|
| 46 |
doCompile = True
|
| 47 |
if doCompile:
|
| 48 |
+
from sfast.compilers.diffusion_pipeline_compiler import (compile, compile_unet, compile_vae, CompilationConfig)
|
| 49 |
config = CompilationConfig.Default()
|
| 50 |
|
| 51 |
try:
|