Update forge_reference.py
This commit is contained in:
parent
3aff8a5c64
commit
c0dc02c488
@ -47,8 +47,18 @@ class PreprocessorReference(Preprocessor):
|
|||||||
|
|
||||||
return h
|
return h
|
||||||
|
|
||||||
|
def attn1_proc(q, k, v, transformer_options):
|
||||||
|
sigma = transformer_options["sigmas"][0].item()
|
||||||
|
if not (sigma_min < sigma < sigma_max):
|
||||||
|
return q, k, k
|
||||||
|
|
||||||
|
a = 0
|
||||||
|
|
||||||
|
return q, k, k
|
||||||
|
|
||||||
unet.add_block_modifier(block_proc)
|
unet.add_block_modifier(block_proc)
|
||||||
unet.add_conditioning_modifier(conditioning_modifier)
|
unet.add_conditioning_modifier(conditioning_modifier)
|
||||||
|
unet.set_model_attn1_patch(attn1_proc)
|
||||||
|
|
||||||
process.sd_model.forge_objects.unet = unet
|
process.sd_model.forge_objects.unet = unet
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user