HuggingFaceFW/fineweb-edu
Viewer • Updated • 3.5B • 513k • 1.13k
How to use kz919/llama3_1b_100B_token_8222025 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="kz919/llama3_1b_100B_token_8222025") # Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("kz919/llama3_1b_100B_token_8222025")
model = AutoModelForCausalLM.from_pretrained("kz919/llama3_1b_100B_token_8222025")How to use kz919/llama3_1b_100B_token_8222025 with vLLM:
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "kz919/llama3_1b_100B_token_8222025"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "kz919/llama3_1b_100B_token_8222025",
"prompt": "Once upon a time,",
"max_tokens": 512,
"temperature": 0.5
}'docker model run hf.co/kz919/llama3_1b_100B_token_8222025
How to use kz919/llama3_1b_100B_token_8222025 with SGLang:
# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
--model-path "kz919/llama3_1b_100B_token_8222025" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "kz919/llama3_1b_100B_token_8222025",
"prompt": "Once upon a time,",
"max_tokens": 512,
"temperature": 0.5
}'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 "kz919/llama3_1b_100B_token_8222025" \
--host 0.0.0.0 \
--port 30000
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:30000/v1/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "kz919/llama3_1b_100B_token_8222025",
"prompt": "Once upon a time,",
"max_tokens": 512,
"temperature": 0.5
}'How to use kz919/llama3_1b_100B_token_8222025 with Docker Model Runner:
docker model run hf.co/kz919/llama3_1b_100B_token_8222025
This model is a 1B llama3 model pretrained from scratch with torchtitan on fineweb-edu with AdamW optimizer. 100B tokens seen.
import torch
from transformers import pipeline
pipe = pipeline(
"text-generation",
model="kz919/llama3_1b_cautious_100B_token_8222025",
)
print(pipe("The key to life is"))
lm_eval --model hf --model_args pretrained=kz919/llama3_1b_cautious_100B_token_8222025,dtype="bfloat16",add_bos_token=True --tasks lambada_openai,hellaswag,piqa,arc_easy,arc_challenge,openbookqa --device cuda:7 --batch_size 8
| Tasks | Version | Filter | n-shot | Metric | Value | Stderr | ||
|---|---|---|---|---|---|---|---|---|
| arc_challenge | 1 | none | 0 | acc | ↑ | 0.3123 | ± | 0.0135 |
| none | 0 | acc_norm | ↑ | 0.3413 | ± | 0.0139 | ||
| arc_easy | 1 | none | 0 | acc | ↑ | 0.6768 | ± | 0.0096 |
| none | 0 | acc_norm | ↑ | 0.5922 | ± | 0.0101 | ||
| hellaswag | 1 | none | 0 | acc | ↑ | 0.4007 | ± | 0.0049 |
| none | 0 | acc_norm | ↑ | 0.5092 | ± | 0.0050 | ||
| lambada_openai | 1 | none | 0 | acc | ↑ | 0.3901 | ± | 0.0068 |
| none | 0 | perplexity | ↓ | 21.6290 | ± | 0.7689 | ||
| openbookqa | 1 | none | 0 | acc | ↑ | 0.2660 | ± | 0.0198 |
| none | 0 | acc_norm | ↑ | 0.3680 | ± | 0.0216 | ||
| piqa | 1 | none | 0 | acc | ↑ | 0.7127 | ± | 0.0106 |
| none | 0 | acc_norm | ↑ | 0.7100 | ± | 0.0106 |
| Groups | Version | Filter | n-shot | Metric | Value | Stderr | ||
|---|---|---|---|---|---|---|---|---|
| mmlu | 2 | none | acc | ↑ | 0.2515 | ± | 0.0037 | |
| - humanities | 2 | none | acc | ↑ | 0.2451 | ± | 0.0063 | |
| - other | 2 | none | acc | ↑ | 0.2716 | ± | 0.0080 | |
| - social sciences | 2 | none | acc | ↑ | 0.2476 | ± | 0.0078 | |
| - stem | 2 | none | acc | ↑ | 0.2452 | ± | 0.0076 |