Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "7dd352cbd7e8a7dffd47edd0987ef7c91c9924aa"
Unverified Commit 07ef4855 authored by takuoko's avatar takuoko Committed by GitHub
Browse files

[Community, Enhancement] Add reference tricks in README (#3589) 

add reference tricks
parent 6cbddf55
Show whitespace changes
Inline Side-by-side
Showing with 21 additions and 16 deletions
examples/community/README.md
View file @ 07ef4855
...@@ -1326,6 +1326,8 @@ image.save('tensorrt_img2img_new_zealand_hills.png') ...@@ -1326,6 +1326,8 @@ image.save('tensorrt_img2img_new_zealand_hills.png')
This pipeline uses the Reference Control. Refer to the [sd-webui-controlnet discussion: Reference-only Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1236)[sd-webui-controlnet discussion: Reference-adain Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1280). This pipeline uses the Reference Control. Refer to the [sd-webui-controlnet discussion: Reference-only Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1236)[sd-webui-controlnet discussion: Reference-adain Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1280).
Based on [this issue](https://github.com/huggingface/diffusers/issues/3566),
- `EulerAncestralDiscreteScheduler` got poor results.
```py ```py
import torch import torch
...@@ -1369,6 +1371,9 @@ Output Image of `reference_attn=True` and `reference_adain=True` ...@@ -1369,6 +1371,9 @@ Output Image of `reference_attn=True` and `reference_adain=True`
This pipeline uses the Reference Control with ControlNet. Refer to the [sd-webui-controlnet discussion: Reference-only Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1236)[sd-webui-controlnet discussion: Reference-adain Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1280). This pipeline uses the Reference Control with ControlNet. Refer to the [sd-webui-controlnet discussion: Reference-only Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1236)[sd-webui-controlnet discussion: Reference-adain Control](https://github.com/Mikubill/sd-webui-controlnet/discussions/1280).
Based on [this issue](https://github.com/huggingface/diffusers/issues/3566),
- `EulerAncestralDiscreteScheduler` got poor results.
- `guess_mode=True` works well for ControlNet v1.1
```py ```py
import cv2 import cv2
......
examples/community/stable_diffusion_controlnet_reference.py
View file @ 07ef4855
...@@ -505,8 +505,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -505,8 +505,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -545,8 +545,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -545,8 +545,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -605,8 +605,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -605,8 +605,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -642,8 +642,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli ...@@ -642,8 +642,8 @@ class StableDiffusionControlNetReferencePipeline(StableDiffusionControlNetPipeli
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
......
examples/community/stable_diffusion_reference.py
View file @ 07ef4855
...@@ -499,8 +499,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline): ...@@ -499,8 +499,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline):
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -539,8 +539,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline): ...@@ -539,8 +539,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline):
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -599,8 +599,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline): ...@@ -599,8 +599,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline):
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
...@@ -636,8 +636,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline): ...@@ -636,8 +636,8 @@ class StableDiffusionReferencePipeline(StableDiffusionPipeline):
if MODE == "write": if MODE == "write":
if gn_auto_machine_weight >= self.gn_weight: if gn_auto_machine_weight >= self.gn_weight:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
self.mean_bank.append(mean) self.mean_bank.append([mean])
self.var_bank.append(var) self.var_bank.append([var])
if MODE == "read": if MODE == "read":
if len(self.mean_bank) > 0 and len(self.var_bank) > 0: if len(self.mean_bank) > 0 and len(self.var_bank) > 0:
var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0) var, mean = torch.var_mean(hidden_states, dim=(2, 3), keepdim=True, correction=0)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment