Commit 1f6bfdea authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf

move the modified decode into smapler_common

parent 70e66e81
......@@ -55,9 +55,9 @@ def single_sample_to_image(sample, approximation=None):
def decode_first_stage(model, x):
x = model.decode_first_stage(x.to(devices.dtype_vae))
return x
x = x.to(devices.dtype_vae)
approx_index = approximation_indexes.get(opts.sd_vae_decode_method, 0)
return samples_to_images_tensor(x, approx_index, model)
def sample_to_image(samples, index=0, approximation=None):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment