Fix uni3c reloading

This commit is contained in:
kijai
2025-12-14 16:48:55 +02:00
parent 0fa5383106
commit 3611341339
3 changed files with 6 additions and 6 deletions
+1 -1
View File
@@ -890,7 +890,7 @@ class WanVideoSampler:
#uni3c
uni3c_data = uni3c_data_input = None
if uni3c_embeds is not None:
transformer.controlnet = uni3c_embeds["controlnet"]
transformer.uni3c_controlnet = uni3c_embeds["controlnet"]
render_latent = uni3c_embeds["render_latent"].to(device)
if render_latent.shape != noise.shape:
render_latent = torch.nn.functional.interpolate(render_latent, size=(noise.shape[1], noise.shape[2], noise.shape[3]), mode='trilinear', align_corners=False)