Loading…
Loading…
Luna 3.1 is our next generation text model designed for heavy usage and lengthy conversations. These models have added Llama.cpp support as well as faster response time. Luna has an 8k context window, which supports long chats and roleplays. Luna 3.1 retains the Lightning architecture, with the only changes being upgraded RMSNorm and RoPE. Luna 3.1 generates at a relatively fast speed of 130 tokens per second.
[!IMPORTANT] If you try installing llama.cpp traditionally, the program will exit with
unknown model architecture: 'lightning'. A patch of llama.cpp has been released called Lightning.cpp that supports the Luna architecture. Lightning.cpp
Luna can be used directly for text generation, experimentation, and conversational interactions. Users can provide text prompts and generate responses using the model's built-in language modeling capabilities. Direct use is primarily intended for casual conversation, research and experimentation. Outputs may be incomplete, inaccurate, repetitive, or unrelated to the input, and should be evaluated before being used for other purposes.
We recommend users of Luna to fine-tune the model on new text, and add necessary guardrails and precautions to prevent misuse.
```python
from transformers import pipeline
question = "Replace with your question"
generator = pipeline("text-generation", model="None", device_map="auto")
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
print(output["generated_text"])
Luna was trained on a subset of the OASST dataset.
Luna 3 was trained on an RTX 3050 GPU, using FlashAttention/SDPA and MHA. The model was trained on Trl's capybara dataset.
| Hyperparameter | Value | Comment |
|---|---|---|
| Precision | BF16 | |
| Optimizer | AdamW | |
| Learning rate | 2e-5 | |
| Batch size | 17 |
Luna uses a 12-layer causal Transformer with 768-dimensional hidden states and 12 attention heads, with RMSNorm and RoPE. Luna uses a 8126 context window and Multi-Head attention.
| Hyperparameter | Value | Comment |
|---|---|---|
| Layers | 12 | |
| D_MODEL | 768 | |
| Attention Heads | 12 | |
| Vocabulary | ~100k | w/ 8126 Sequence length |
@software{vonwerra2020trl,
title = {{TRL: Transformers Reinforcement Learning}},
author = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
license = {Apache-2.0},
url = {https://github.com/huggingface/trl},
year = {2020}
}
Aobangaming/Luna-3-230m