|
|
--- |
|
|
license: mit |
|
|
language: |
|
|
- en |
|
|
pipeline_tag: text-classification |
|
|
--- |
|
|
# Gemma fine-tuned to Multi-NLI (MNLI |
|
|
This model is for Textual Entailment (NLI), i.e., predict whether sentence 1 is supported by sentence 2. It's a 3-way classification where the relationship between sentence 1 and sentence 2 can be entailment, neutral, contradiction. |
|
|
|
|
|
## Model Details |
|
|
- Base model: [gemma-7b-aps-it](https://huggingface.co/google/gemma-7b-aps-it) |
|
|
- Training data: [MNLI](https://huggingface.co/datasets/multi_nli) |
|
|
- Training details: QLoRA fine tuning, num_epochs = 3 |
|
|
|
|
|
|