diff --git a/comfy/supported_models.py b/comfy/supported_models.py index 185d29299..f0e5ec512 100644 --- a/comfy/supported_models.py +++ b/comfy/supported_models.py @@ -911,7 +911,7 @@ class WAN21_T2V(supported_models_base.BASE): memory_usage_factor = 1.0 - supported_inference_dtypes = [torch.bfloat16, torch.float32] + supported_inference_dtypes = [torch.bfloat16, torch.float16, torch.float32] vae_key_prefix = ["vae."] text_encoder_key_prefix = ["text_encoders."] diff --git a/comfy_extras/nodes_video.py b/comfy_extras/nodes_video.py index f3922e03d..53920ba18 100644 --- a/comfy_extras/nodes_video.py +++ b/comfy_extras/nodes_video.py @@ -59,6 +59,7 @@ class SaveWEBM: frame = av.VideoFrame.from_ndarray(torch.clamp(frame[..., :3] * 255, min=0, max=255).to(device=torch.device("cpu"), dtype=torch.uint8).numpy(), format="rgb24") for packet in stream.encode(frame): container.mux(packet) + container.mux(stream.encode()) container.close() results = [{