CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
Haoming 2026-02-10 16:49:42 +08:00
parent d4894dd1f1
commit db6d3ab26e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/diffusion_engine/wan.py
View File

@ -104,11 +104,11 @@ class Wan(ForgeDiffusionEngine):
@torch.inference_mode()
def encode_first_stage(self, x: torch.Tensor):
b, c, h, w = x.shape
if x.size(0) > 1:
x = x[0].unsqueeze(0) # enforce batch_size of 1
start_image = x.movedim(1, -1) * 0.5 + 0.5
b, c, h, w = x.shape
latent = torch.zeros([1, 16, ((b - 1) // 4) + 1, h // 8, w // 8], device=self.forge_objects.vae.device)
self.image_to_video(b, start_image, latent)
sample = self.forge_objects.vae.first_stage_model.process_in(latent)