lamatama / README.md
kevin009's picture
update
bb8349c verified
|
raw
history blame
3.08 kB
metadata
license: apache-2.0
language:
  - en

Model Card: kevin009/lamatama

Model Description

The kevin009/lamatama model is a groundbreaking achievement in the field of language modeling, showcasing the power of leveraging a substantial dataset and state-of-the-art training techniques. This model is designed to push the boundaries of what's possible in natural language understanding and generation.

Training Details

  • Model Architecture: The kevin009/lamatama model is built upon the architecture and tokenizer of Llama 2, ensuring compatibility and easy integration with various open-source projects.
  • Dataset: It was pretrained on an impressive 3 trillion tokens, a scale that allows for a deep and nuanced understanding of language.
  • Training Period: The training process was carried out over 90 days, utilizing 16 A100-40G GPUs, a testament to the model's efficiency and the team's optimization skills.

Fine-tuning

This specific version of the model has been fine-tuned to excel in chat-based applications. It builds upon the TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T model, incorporating learnings and optimizations from HF's Zephyr's training recipe.

  • Initial Phase: The model was first fine-tuned on a variant of the UltraChat dataset, which is rich in synthetic dialogues generated by ChatGPT.
  • Further Alignment: Subsequent alignment was achieved using 🤗 TRL's DPOTrainer with the openbmb/UltraFeedback dataset, comprising 64k prompts and model completions ranked by GPT-4.

How to Use

Ensure you have transformers>=4.34. For detailed instructions and updates, check out the GitHub page for kevin009/lamatama.

Installation (for versions <= v4.34)

pip install git+https://github.com/huggingface/transformers.git
pip install accelerate

Example Usage

Here's a quick guide on using kevin009/lamatama for generating text:

import torch
from transformers import pipeline

# Initialize the pipeline
pipe = pipeline("text-generation", model="kevin009/lamatama", torch_dtype=torch.bfloat16, device_map="auto")

# Sample dialogue with templating
messages = [
    {"role": "system", "content": "You are a friendly chatbot who always responds in the style of a pirate"},
    {"role": "user", "content": "How many helicopters can a human eat in one sitting?"}
]

# Generate prompt and outputs
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
print(outputs[0]["generated_text"])

Acknowledgements

This model is a product of collaboration and innovative approaches to language modeling. We extend our thanks to all contributors, as well as the creators of the datasets and training methodologies that made kevin009/lamatama a reality.


This model card introduces kevin009/lamatama, a versatile and powerful language model fine-tuned for chat applications, demonstrating exceptional understanding and generation capabilities.