Instructions to use Mongjin/ft_memory_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Mongjin/ft_memory_model with Transformers:
# Load model directly from transformers import AutoTokenizer, BartForDialogueGeneration tokenizer = AutoTokenizer.from_pretrained("Mongjin/ft_memory_model") model = BartForDialogueGeneration.from_pretrained("Mongjin/ft_memory_model", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 191 Bytes
8f61bcd | 1 | {"bos_token": "</s>", "eos_token": "</s>", "unk_token": "<unk>", "pad_token": "<pad>", "mask_token": "<mask>", "additional_special_tokens": ["<agent>", "<user>", "<agent_mem>", "<user_mem>"]} |