minor fixes

This commit is contained in:
Vladimir Mandic
2023-06-03 10:36:53 -04:00
parent 4265692505
commit 549c02a874
4 changed files with 2 additions and 3 deletions
-1
View File
@@ -45,7 +45,6 @@ function checkPaused(state) {
}
function setProgress(res) {
console.log(res);
elements = ['txt2img_generate', 'img2img_generate', 'extras_generate']
const progress = (res?.progress || 0)
const perc = res && (progress > 0) ? `${Math.round(100.0 * progress)}%` : ''
-1
View File
@@ -434,7 +434,6 @@ def atomically_save_image():
filename = filename.strip()
if extension[0] != '.': # add dot if missing
extension = '.' + extension
print('image_format', extension)
try:
image_format = Image.registered_extensions()[extension]
except Exception:
+1 -1
View File
@@ -922,7 +922,7 @@ class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):
decoded_samples = torch.from_numpy(np.array(batch_images))
decoded_samples = decoded_samples.to(shared.device)
decoded_samples = 2. * decoded_samples - 1.
if shared.opts.sd_vae_sliced_encode:
if shared.opts.sd_vae_sliced_encode and len(decoded_samples) > 1:
samples = torch.stack([
self.sd_model.get_first_stage_encoding(
self.sd_model.encode_first_stage(torch.unsqueeze(decoded_sample, 0))
+1
View File
@@ -73,6 +73,7 @@ def apply_checkpoint(p, x, xs):
shared.log.warning(f"XYZ grid: unknown checkpoint: {x}")
else:
sd_models.reload_model_weights(shared.sd_model, info)
p.override_settings['sd_model_checkpoint'] = info.name
def confirm_checkpoints(p, xs):