update

README.md

@@ -23,11 +23,11 @@ from transformers import AutoModel, CLIPImageProcessor
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model = AutoModel.from_pretrained(
-    'jienengchen/ViTamin-L-336px',
+    'jienengchen/ViTamin-B-LTT',
     trust_remote_code=True).to(device).eval()
 
 image = Image.open('./image.png').convert('RGB')
-image_processor = CLIPImageProcessor.from_pretrained('jienengchen/ViTamin-L-336px')
+image_processor = CLIPImageProcessor.from_pretrained('jienengchen/ViTamin-B-LTT')
 
 pixel_values = image_processor(images=image, return_tensors='pt').pixel_values
 pixel_values = pixel_values.to(torch.bfloat16).cuda()