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