vegetarian-mayo / README.md
nroggendorff's picture
Update README.md
a94c217 verified
|
raw
history blame
1.03 kB
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.