---
license: apache-2.0
language:
- it
library_name: peft
pipeline_tag: text-generation
tags:
- legal
base_model: mistralai/Mistral-7B-Instruct-v0.1
---

# Model Description

A Mistral-7B-instruct-v0.1 model to extract a title from the text of Italian law articles. It is fine-tuned over a set of 100k text-title pairs that are available throughout the Italian legislation. It can be used to extract titles for articles or attachments that do not have a pre-defined title.


- **Developed by:** Andrea Colombo, Politecnico di Milano
- **Model type:** text generation
- **Language(s) (NLP):** Italian
- **License:** Apache 2.0
- **Finetuned from model:** mistralai/Mistral-7B-Instruct-v0.1


## Training Details

### Training Procedure

The model has been trained for 100 training steps with batch size 4, 4-bit quantization using bitsandbytes and a LoRA rank of 64.
We use the paged Adam optimizer, a learning rate of 0.004, and a cosine learning rate scheduler with a 0.03 warm-up fraction.


## Evaluation

The best model reported an evaluation loss of 1.0030452013015747