Transformers How to use fatehmujtaba/flan-t5-Small-for-Chest-Xray with Transformers:
# Load model directly
from transformers import AutoTokenizer, CustomVisionT5
tokenizer = AutoTokenizer.from_pretrained("fatehmujtaba/flan-t5-Small-for-Chest-Xray")
model = CustomVisionT5.from_pretrained("fatehmujtaba/flan-t5-Small-for-Chest-Xray", device_map="auto") 
