Update README.md

README.md

@@ -190,7 +190,7 @@ import torch
 from cosmos_tokenizer.image_lib import ImageTokenizer
 
 model_name = "Cosmos-Tokenizer-CI8x8"
-input_tensor = torch.randn(1, 3, 512, 512).to('cuda').to(torch.bfloat16)  # [B, C, T, H, W]
+input_tensor = torch.randn(1, 3, 512, 512).to('cuda').to(torch.bfloat16)  # [B, C, H, W]
 encoder = ImageTokenizer(checkpoint_enc=f'pretrained_ckpts/{model_name}/encoder.jit')
 (latent,) = encoder.encode(input_tensor)
 torch.testing.assert_close(latent.shape, (1, 16, 64, 64))