soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

i

Browse Source
This commit is contained in:
lllyasviel 2024-01-25 21:01:30 -08:00
parent c6370ff074
commit a0e3b71fab
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_cfg_denoiser.py
View File

@ -218,6 +218,8 @@ class CFGDenoiser(torch.nn.Module):
tensor = tensor.to(unet_dtype)
uncond = uncond.to(unet_dtype)
self.inner_model.inner_model.current_sigmas = sigma_in
if tensor.shape[1] == uncond.shape[1] or skip_uncond:
if is_edit_model:
cond_in = catenate_conds([tensor, uncond, uncond])

2
modules_forge/forge_loader.py
View File

@ -252,7 +252,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None):
def forge_unet_forward(*args, **kwargs):
current_transformer_options = kwargs.get('transformer_options', {})
current_transformer_options.update(dict(cond_or_uncond=sd_model.cond_or_uncond))
current_transformer_options.update(dict(cond_or_uncond=sd_model.cond_or_uncond, sigmas=sd_model.current_sigmas))
current_transformer_options.update(sd_model.forge_objects.unet.model_options.get('transformer_options', {}))
kwargs.update(dict(