Skip to content

Commit d53868a

Browse files
2 parents c7e9d6e + a6e13fb commit d53868a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_oneflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ class UNetGraph(torch.nn.Graph):
4040
def __init__(self, unet):
4141
super().__init__()
4242
self.unet = unet
43-
self.config.enable_cudnn_conv_heuristic_search_algo(False)
44-
self.config.allow_fuse_add_to_output(False)
43+
self.config.enable_cudnn_conv_heuristic_search_algo(True)
44+
self.config.allow_fuse_add_to_output(True)
4545

4646
def build(self, latent_model_input, t, text_embeddings):
4747
text_embeddings = torch._C.amp_white_identity(text_embeddings)

0 commit comments

Comments
 (0)