Downloads · 30 days
33
18% of all-time downloads
PRYMMAL-6B-slerp
LilRg/PRYMMAL-6B-slerp
PRYMMAL-6B-slerp is a machine learning model from LilRg. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
PRYMMAL-6B-slerp is a merge of the following models using LazyMergekit: 01-ai/Yi-1.5-6B-Chat 01-ai/Yi-1.5-6B
All-time downloads
184
Public
Parameters
3.3B
6.6 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors6.6 GB · 100%
At a glance
- License
- apache-2.0
- Model type
- llama
- Access
- Public
- Created
- Sep 24, 2024
- Updated
- Sep 24, 2024
- SHA
- 1ce0f5fd
Base models
Model card
From the Hugging Face model README
PRYMMAL-6B-slerp
PRYMMAL-6B-slerp is a merge of the following models using LazyMergekit:
🧩 Configuration
slices:
- sources:
- model: 01-ai/Yi-1.5-6B-Chat
layer_range: [0, 16]
- model: 01-ai/Yi-1.5-6B
layer_range: [16, 32]
merge_method: slerp
base_model: 01-ai/Yi-1.5-6B-Chat
parameters:
t:
- filter: self_attn
value: [0, 0.25, 0.5, 0.75, 1]
- filter: mlp
value: [1, 0.75, 0.5, 0.25, 0]
- value: 0.5
dtype: bfloat16
💻 Usage
!pip install -qU transformers accelerate
from transformers import AutoTokenizer
import transformers
import torch
model = "LilRg/PRYMMAL-6B-slerp"
messages = [{"role": "user", "content": "What is a large language model?"}]
tokenizer = AutoTokenizer.from_pretrained(model)
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
pipeline = transformers.pipeline(
"text-generation",
model=model,
torch_dtype=torch.float16,
device_map="auto",
)
outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
print(outputs[0]["generated_text"])
- Type
- llama
- License
- apache-2.0
- Created
- Sep 24, 2024
- Updated
- Sep 24, 2024