add reference

Blizaine · Nov 8, 2023 · e31d3e1 · e31d3e1
1 parent 1af368e
commit e31d3e1
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/app-controlnet.py b/app-controlnet.py
@@ -107,8 +107,8 @@
     truncate_long_prompts=False,
 )
 if TORCH_COMPILE:
-    pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
-    pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
+    pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+    pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
 
     pipe(prompt="warmup", image=[Image.new("RGB", (768, 768))], control_image=[Image.new("RGB", (768, 768))])
 

diff --git a/app-img2img.py b/app-img2img.py
@@ -79,8 +79,8 @@
     pipe.enable_attention_slicing()
 
 if TORCH_COMPILE:
-    pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
-    pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
+    pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+    pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
 
     pipe(prompt="warmup", image=[Image.new("RGB", (512, 512))])
 

diff --git a/app-txt2img.py b/app-txt2img.py
@@ -79,8 +79,8 @@
     pipe.enable_attention_slicing()
 
 if TORCH_COMPILE:
-    pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=False)
-    pipe.vae = torch.compile(pipe.vae, mode="max-autotune", fullgraph=False)
+    pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+    pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
 
     pipe(prompt="warmup", num_inference_steps=1, guidance_scale=8.0)
 

diff --git a/latent_consistency_controlnet.py b/latent_consistency_controlnet.py
@@ -1,3 +1,4 @@
+# from https://github.com/taabata/LCM_Inpaint_Outpaint_Comfy/blob/main/LCM/pipeline_cn.py
 # Copyright 2023 Stanford University Team and The HuggingFace Team. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");