cosmos predict2 text to image 2B and 14B models not working.

[cagri-comfyui]

Custom Node Testing

• [x] I have tried disabling custom nodes and the issue persists (see how to disable custom nodes if you need help)

Your question

ERROR: Could not detect model type of: (PATH)\comfyui_NEW\ComfyUI_windows_portable\ComfyUI\models\diffusion_models\cosmos_t2i.pt latest version on comfyui using rtx 3090

Logs

Other

No response

[kanttouchthis]

did you run update_comfyui.bat? cosmos support was added added after v0.3.40 and there hasn't been a new release since.

[jav12z]

I've just tried cosmos predict 2B img2video model-480p-16fps.pt, using the new node and it's giving me errors loading the VAE.

Error(s) in loading state_dict for AutoencoderKL: size mismatch for encoder.conv_in.weight: copying a param with shape torch.Size([96, 3, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 3, 3, 3]). size mismatch for encoder.conv_in.bias: copying a param with shape torch.Size([96]) from checkpoint, the shape in current model is torch.Size([128]). size mismatch for encoder.conv_out.weight: copying a param with shape torch.Size([32, 384, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([32, 512, 3, 3]). size mismatch for decoder.conv_in.weight: copying a param with shape torch.Size([384, 16, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 16, 3, 3]). size mismatch for decoder.conv_in.bias: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([512]). size mismatch for decoder.conv_out.weight: copying a param with shape torch.Size([3, 96, 3, 3, 3]) from checkpoint, the shape in current model is torch.Size([3, 128, 3, 3]). size mismatch for quant_conv.weight: copying a param with shape torch.Size([32, 32, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 32, 1, 1]). size mismatch for post_quant_conv.weight: copying a param with shape torch.Size([16, 16, 1, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 16, 1, 1]).

[cagri-comfyui]

sorry for the late reply. i did update via .bat file and it worked... thanks

[poluramus]

Is there any workflow for the new video model ?

[github-actions[bot]]

This issue is being marked stale because it has not had any activity for 30 days. Reply below within 7 days if your issue still isn't solved, and it will be left open. Otherwise, the issue will be closed automatically.