support loading .yaml config with same name as model

support EMA weights in processing (????)
This commit is contained in:
AUTOMATIC
2022-10-08 23:26:48 +03:00
parent 432782163a
commit 050a6a798c
2 changed files with 24 additions and 8 deletions

View File

@@ -347,7 +347,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
infotexts = []
output_images = []
with torch.no_grad():
with torch.no_grad(), p.sd_model.ema_scope():
with devices.autocast():
p.init(all_prompts, all_seeds, all_subseeds)