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
| {"additional_special_tokens": ["<agent>", "<user>", "<agent_mem>", "<user_mem>"], "special_tokens_map_file": "/home/mongjin/.cache/huggingface/transformers/a87d2ed77831bb40ce806a97c04126addf5ecc82b3e23ecf916b2a4acdb9c29a.c23d5e62137984cf842a885705037b25b156747d145406702932d5f5d5e7c88e", "name_or_path": "gogamza/kobart-base-v2", "tokenizer_class": "PreTrainedTokenizerFast"} |