Spaces:
Sleeping
Sleeping
rynmurdock
commited on
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)
|