Update README.md

Fix a mistake in a comment in the example usage code

README.md

@@ -29,7 +29,7 @@ untransform_image = transforms.Compose([
 model = AutoencoderKL.from_pretrained("scrumptious/librevae-f8-d8").to("cuda").eval()
 model.requires_grad_(False)
 
-# For a 512x512 image, image_tensor will be (1, 8, 64, 64)
+# For a 512x512 image, image_tensor will be (1, 3, 512, 512)
 image_tensor = transform_image(Image.open("sample.png")).unsqueeze(0).to("cuda")
 # latent will be (1, 8, 64, 64)
 # we multiply it by the scaling factor so it has an approximate mean of 0 and variance of 1