checkpoint = "openai/clip-vit-large-patch14"
model = CLIPTextModel.from_pretrained(
checkpoint, torch_dtype=torch.bfloat16
)
tokenizer = CLIPTokenizer.from_pretrained(
checkpoint
)
model.push_to_hub("ariG23498/clip-vit-large-patch14-torch")
tokenizer.push_to_hub("ariG23498/clip-vit-large-patch14-torch")