Fix code style

This commit is contained in:
Kohaku-Blueleaf 2023-08-04 13:40:20 +08:00
parent 75336dfc84
commit c134a48016

View File

@ -2,7 +2,7 @@ from collections import namedtuple
import numpy as np import numpy as np
import torch import torch
from PIL import Image from PIL import Image
from modules import devices, processing, images, sd_vae_approx, sd_samplers, sd_vae_taesd, shared from modules import devices, images, sd_vae_approx, sd_samplers, sd_vae_taesd, shared
from modules.shared import opts, state from modules.shared import opts, state
SamplerData = namedtuple('SamplerData', ['name', 'constructor', 'aliases', 'options']) SamplerData = namedtuple('SamplerData', ['name', 'constructor', 'aliases', 'options'])
@ -40,7 +40,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None):
if model is None: if model is None:
model = shared.sd_model model = shared.sd_model
x_sample = model.decode_first_stage(sample) x_sample = model.decode_first_stage(sample)
return x_sample return x_sample