HuggingFaceH4/ultrachat_200k
Viewer • Updated • 515k • 70.9k • 707
How to use Felladrin/Minueza-32M-UltraChat with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="Felladrin/Minueza-32M-UltraChat")
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe(messages) # Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("Felladrin/Minueza-32M-UltraChat")
model = AutoModelForCausalLM.from_pretrained("Felladrin/Minueza-32M-UltraChat")
messages = [
{"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))How to use Felladrin/Minueza-32M-UltraChat with vLLM:
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "Felladrin/Minueza-32M-UltraChat"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "Felladrin/Minueza-32M-UltraChat",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'docker model run hf.co/Felladrin/Minueza-32M-UltraChat
How to use Felladrin/Minueza-32M-UltraChat with SGLang:
# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
--model-path "Felladrin/Minueza-32M-UltraChat" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "Felladrin/Minueza-32M-UltraChat",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'docker run --gpus all \
--shm-size 32g \
-p 30000:30000 \
-v ~/.cache/huggingface:/root/.cache/huggingface \
--env "HF_TOKEN=<secret>" \
--ipc=host \
lmsysorg/sglang:latest \
python3 -m sglang.launch_server \
--model-path "Felladrin/Minueza-32M-UltraChat" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "Felladrin/Minueza-32M-UltraChat",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'How to use Felladrin/Minueza-32M-UltraChat with Docker Model Runner:
docker model run hf.co/Felladrin/Minueza-32M-UltraChat
<|im_start|>system
{system_message}<|im_end|>
<|im_start|>user
{user_message}<|im_end|>
<|im_start|>assistant
do_sample: true
temperature: 0.65
top_p: 0.55
top_k: 35
repetition_penalty: 1.176
from transformers import pipeline
generate = pipeline("text-generation", "Felladrin/Minueza-32M-UltraChat")
messages = [
{
"role": "system",
"content": "You are a highly knowledgeable and friendly assistant. Your goal is to understand and respond to user inquiries with clarity. Your interactions are always respectful, helpful, and focused on delivering the most accurate information to the user.",
},
{
"role": "user",
"content": "Hey! Got a question for you!",
},
{
"role": "assistant",
"content": "Sure! What's it?",
},
{
"role": "user",
"content": "What are some potential applications for quantum computing?",
},
]
prompt = generate.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
output = generate(
prompt,
max_new_tokens=256,
do_sample=True,
temperature=0.65,
top_k=35,
top_p=0.55,
repetition_penalty=1.176,
)
print(output[0]["generated_text"])
This model was trained with SFTTrainer using the following settings:
| Hyperparameter | Value |
|---|---|
| Learning rate | 2e-5 |
| Total train batch size | 16 |
| Max. sequence length | 2048 |
| Weight decay | 0 |
| Warmup ratio | 0.1 |
| Optimizer | Adam with betas=(0.9,0.999) and epsilon=1e-08 |
| Scheduler | cosine |
| Seed | 42 |
Detailed results can be found here
| Metric | Value |
|---|---|
| Avg. | 28.97 |
| AI2 Reasoning Challenge (25-Shot) | 21.08 |
| HellaSwag (10-Shot) | 26.95 |
| MMLU (5-Shot) | 26.08 |
| TruthfulQA (0-shot) | 47.70 |
| Winogrande (5-shot) | 51.78 |
| GSM8k (5-shot) | 0.23 |
docker model run hf.co/Felladrin/Minueza-32M-UltraChat