Spaces:
Runtime error
Runtime error
Commit
·
c1c2061
1
Parent(s):
b4057bb
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,21 +15,6 @@ import matplotlib.pyplot as plt
|
|
| 15 |
import matplotlib
|
| 16 |
|
| 17 |
from sklearn.linear_model import Ridge
|
| 18 |
-
from sfast.compilers.diffusion_pipeline_compiler import (compile, compile_unet,
|
| 19 |
-
CompilationConfig)
|
| 20 |
-
config = CompilationConfig.Default()
|
| 21 |
-
|
| 22 |
-
try:
|
| 23 |
-
import triton
|
| 24 |
-
config.enable_triton = True
|
| 25 |
-
except ImportError:
|
| 26 |
-
print('Triton not installed, skip')
|
| 27 |
-
config.enable_cuda_graph = True
|
| 28 |
-
config.enable_jit = True
|
| 29 |
-
config.enable_jit_freeze = True
|
| 30 |
-
config.enable_cnn_optimization = True
|
| 31 |
-
config.preserve_parameters = False
|
| 32 |
-
config.prefer_lowp_gemm = True
|
| 33 |
|
| 34 |
import imageio
|
| 35 |
import gradio as gr
|
|
@@ -126,7 +111,6 @@ pipe.set_ip_adapter_scale(.8)
|
|
| 126 |
pipe.unet.fuse_qkv_projections()
|
| 127 |
#pipe.enable_free_init(method="gaussian", use_fast_sampling=True)
|
| 128 |
|
| 129 |
-
#pipe = compile(pipe, config=config)
|
| 130 |
pipe.to(device=DEVICE)
|
| 131 |
#pipe.unet = torch.compile(pipe.unet)
|
| 132 |
#pipe.vae = torch.compile(pipe.vae)
|
|
|
|
| 15 |
import matplotlib
|
| 16 |
|
| 17 |
from sklearn.linear_model import Ridge
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
import imageio
|
| 20 |
import gradio as gr
|
|
|
|
| 111 |
pipe.unet.fuse_qkv_projections()
|
| 112 |
#pipe.enable_free_init(method="gaussian", use_fast_sampling=True)
|
| 113 |
|
|
|
|
| 114 |
pipe.to(device=DEVICE)
|
| 115 |
#pipe.unet = torch.compile(pipe.unet)
|
| 116 |
#pipe.vae = torch.compile(pipe.vae)
|