lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-21 07:13:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

vae already sliced in inner loop

Browse Source
This commit is contained in:
lllyasviel
2024-03-08 00:40:33 -08:00
parent e48533bdcd
commit 10b5ca2541
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/processing.py
View File

@@ -633,10 +633,10 @@ class DecodedSamples(list):
def decode_latent_batch(model, batch, target_device=None, check_for_nans=False):
samples = DecodedSamples()
samples_pytorch = decode_first_stage(model, batch).to(target_device)
for i in range(batch.shape[0]):
sample = decode_first_stage(model, batch[i:i + 1])[0]
samples.append(sample.to(target_device))
for x in samples_pytorch:
samples.append(x)
return samples