Update README.md

README.md

@@ -33,14 +33,14 @@ def crop_image_to_nearest_divisible_by_8(img):
         # Calculate the closest lower resolution divisible by 8
         new_height = img.shape[1] - (img.shape[1] % 8)
         new_width = img.shape[2] - (img.shape[2] % 8)
-
+        
         # Use CenterCrop to crop the image
-        transform = transforms.Compose([transforms.CenterCrop((new_height, new_width)) , t>
+        transform = transforms.Compose([transforms.CenterCrop((new_height, new_width)) , transforms.ToTensor])
         img = transform(img).to(torch.float32).clamp(-1, 1)
-
+        
         return img
 
-vae = AsymmetricAutoencoderKL.from_pretrained("Heasterian/AsymmetricAutoencoderKLUpscaler">
+vae = AsymmetricAutoencoderKL.from_pretrained("Heasterian/AsymmetricAutoencoderKLUpscaler", weight_dtype=torch.float32)
 vae.requires_grad_(False)
 
 image = load_image(r"/home/heasterian/test/a/F8VlGmCWEAAUVpc (copy).jpeg")