README.md

metadata

language:
  - en
license: mit
tags:
  - trl
  - sft
  - sgd
base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
datasets:
  - nroggendorff/mayo
model-index:
  - name: mayo
    results: []

Mayonnaise LLM

Mayo is a language model fine-tuned on the Mayo dataset using Supervised Fine-Tuning (SFT) and Teacher Reinforced Learning (TRL) techniques. It is based on the TinyLlama Model

Features

• Utilizes SFT and TRL techniques for improved performance
• Supports English language

Usage

To use the Mayo LLM, you can load the model using the Hugging Face Transformers library:

from transformers import pipeline

pipe = pipeline("text-generation", model="nroggendorff/vegetarian-mayo")

question = "What color is the sky?"
conv = [{"role": "user", "content": question}]

response = pipe(conv, max_new_tokens=32)[0]['generated_text'][-1]['content']
print(response)

License

This project is licensed under the MIT License.