| library_name: peft | |
| base_model: mistralai/Mistral-7B-v0.1 | |
| pipeline_tag: text-generation | |
| Description: Topic extraction from a news article and title\ | |
| Original dataset: https://huggingface.co/datasets/fancyzhx/dbpedia_14 \ | |
| ---\ | |
| Try querying this adapter for free in Lora Land at https://predibase.com/lora-land! \ | |
| The adapter_category is Topic Identification and the name is News Topic Identification (dbpedia)\ | |
| ---\ | |
| Sample input: You are given the title and the body of an article below. Please determine the type of the article.\n### Title: Great White Whale\n\n### Body: Great White Whale is the debut album by the Canadian rock band Secret and Whisper. The album was in the works for about a year and was released on February 12 2008. A music video was shot in Pittsburgh for the album's first single XOXOXO. The album reached number 17 on iTunes's top 100 albums in its first week on sale.\n\n### Article Type: \ | |
| ---\ | |
| Sample output: 11\ | |
| ---\ | |
| Try using this adapter yourself! | |
| ``` | |
| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| model_id = "mistralai/Mistral-7B-v0.1" | |
| peft_model_id = "predibase/dbpedia" | |
| model = AutoModelForCausalLM.from_pretrained(model_id) | |
| model.load_adapter(peft_model_id) | |
| ``` |