rynmurdock commited on
Commit
c1c2061
1 Parent(s): b4057bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -16
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)