Views
No views yet
1from transformers import AutoProcessor, AutoModelForImageTextToText
2import torch
3
4model = AutoModelForImageTextToText.from_pretrained(
5 "{REPO_ID}",
6 torch_dtype=torch.float16,
7 device_map="auto",
8 trust_remote_code=True,
9)
10processor = AutoProcessor.from_pretrained("{REPO_ID}", trust_remote_code=True){REPO_ID} with the repo ID of this model.