Transformers How to use PakistanLegalAI/test_arslan with Transformers:
# Load model directly
from transformers import AutoTokenizer, CustomBartForGeneration
tokenizer = AutoTokenizer.from_pretrained("PakistanLegalAI/test_arslan")
model = CustomBartForGeneration.from_pretrained("PakistanLegalAI/test_arslan", device_map="auto") 
