Instructions to use DCAgent/taskmaster2-1k-traces with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use DCAgent/taskmaster2-1k-traces with Transformers:

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="DCAgent/taskmaster2-1k-traces")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)

# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("DCAgent/taskmaster2-1k-traces")
model = AutoModelForCausalLM.from_pretrained("DCAgent/taskmaster2-1k-traces", device_map="auto")
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=256)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))

Notebooks
Google Colab
Kaggle
Local Apps Settings

vLLM

How to use DCAgent/taskmaster2-1k-traces with vLLM:

Install from pip and serve model

# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "DCAgent/taskmaster2-1k-traces"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
	-H "Content-Type: application/json" \
	--data '{
		"model": "DCAgent/taskmaster2-1k-traces",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker

docker model run hf.co/DCAgent/taskmaster2-1k-traces

SGLang

How to use DCAgent/taskmaster2-1k-traces with SGLang:

Install from pip and serve model

# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
    --model-path "DCAgent/taskmaster2-1k-traces" \
    --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": "DCAgent/taskmaster2-1k-traces",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Use Docker images

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 "DCAgent/taskmaster2-1k-traces" \
        --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": "DCAgent/taskmaster2-1k-traces",
		"messages": [
			{
				"role": "user",
				"content": "What is the capital of France?"
			}
		]
	}'

Docker Model Runner
How to use DCAgent/taskmaster2-1k-traces with Docker Model Runner:
```
docker model run hf.co/DCAgent/taskmaster2-1k-traces
```

taskmaster2-1k-traces / trainer_log.jsonl

penfever

Model save

756aaed verified about 1 year ago

Raw History Blame Contribute Delete

3.14 kB

{"current_steps": 5, "total_steps": 80, "loss": 0.743, "lr": 2e-05, "epoch": 0.31746031746031744, "percentage": 6.25, "elapsed_time": "0:02:02", "remaining_time": "0:30:39"}
{"current_steps": 10, "total_steps": 80, "loss": 0.6426, "lr": 3.998096443163716e-05, "epoch": 0.6349206349206349, "percentage": 12.5, "elapsed_time": "0:03:48", "remaining_time": "0:26:38"}
{"current_steps": 15, "total_steps": 80, "loss": 0.5922, "lr": 3.931851652578137e-05, "epoch": 0.9523809523809523, "percentage": 18.75, "elapsed_time": "0:05:33", "remaining_time": "0:24:05"}
{"current_steps": 20, "total_steps": 80, "loss": 0.5513, "lr": 3.774021666356444e-05, "epoch": 1.253968253968254, "percentage": 25.0, "elapsed_time": "0:07:13", "remaining_time": "0:21:40"}
{"current_steps": 25, "total_steps": 80, "loss": 0.5268, "lr": 3.532088886237956e-05, "epoch": 1.5714285714285714, "percentage": 31.25, "elapsed_time": "0:09:03", "remaining_time": "0:19:56"}
{"current_steps": 30, "total_steps": 80, "loss": 0.5194, "lr": 3.217522858017442e-05, "epoch": 1.8888888888888888, "percentage": 37.5, "elapsed_time": "0:10:56", "remaining_time": "0:18:14"}
{"current_steps": 35, "total_steps": 80, "loss": 0.4999, "lr": 2.8452365234813992e-05, "epoch": 2.1904761904761907, "percentage": 43.75, "elapsed_time": "0:12:37", "remaining_time": "0:16:14"}
{"current_steps": 40, "total_steps": 80, "loss": 0.476, "lr": 2.4328792278762058e-05, "epoch": 2.507936507936508, "percentage": 50.0, "elapsed_time": "0:14:24", "remaining_time": "0:14:24"}
{"current_steps": 45, "total_steps": 80, "loss": 0.4743, "lr": 2e-05, "epoch": 2.825396825396825, "percentage": 56.25, "elapsed_time": "0:16:14", "remaining_time": "0:12:37"}
{"current_steps": 50, "total_steps": 80, "loss": 0.4803, "lr": 1.5671207721237945e-05, "epoch": 3.126984126984127, "percentage": 62.5, "elapsed_time": "0:17:57", "remaining_time": "0:10:46"}
{"current_steps": 55, "total_steps": 80, "loss": 0.4741, "lr": 1.1547634765186016e-05, "epoch": 3.4444444444444446, "percentage": 68.75, "elapsed_time": "0:19:43", "remaining_time": "0:08:57"}
{"current_steps": 60, "total_steps": 80, "loss": 0.4612, "lr": 7.824771419825588e-06, "epoch": 3.761904761904762, "percentage": 75.0, "elapsed_time": "0:21:31", "remaining_time": "0:07:10"}
{"current_steps": 65, "total_steps": 80, "loss": 0.4606, "lr": 4.679111137620442e-06, "epoch": 4.063492063492063, "percentage": 81.25, "elapsed_time": "0:23:19", "remaining_time": "0:05:22"}
{"current_steps": 70, "total_steps": 80, "loss": 0.467, "lr": 2.259783336435566e-06, "epoch": 4.380952380952381, "percentage": 87.5, "elapsed_time": "0:25:07", "remaining_time": "0:03:35"}
{"current_steps": 75, "total_steps": 80, "loss": 0.459, "lr": 6.814834742186361e-07, "epoch": 4.698412698412699, "percentage": 93.75, "elapsed_time": "0:27:01", "remaining_time": "0:01:48"}
{"current_steps": 80, "total_steps": 80, "loss": 0.4442, "lr": 1.9035568362844037e-08, "epoch": 5.0, "percentage": 100.0, "elapsed_time": "0:28:36", "remaining_time": "0:00:00"}
{"current_steps": 80, "total_steps": 80, "epoch": 5.0, "percentage": 100.0, "elapsed_time": "0:28:36", "remaining_time": "0:00:00"}

	{"current_steps": 5, "total_steps": 80, "loss": 0.743, "lr": 2e-05, "epoch": 0.31746031746031744, "percentage": 6.25, "elapsed_time": "0:02:02", "remaining_time": "0:30:39"}
	{"current_steps": 10, "total_steps": 80, "loss": 0.6426, "lr": 3.998096443163716e-05, "epoch": 0.6349206349206349, "percentage": 12.5, "elapsed_time": "0:03:48", "remaining_time": "0:26:38"}
	{"current_steps": 15, "total_steps": 80, "loss": 0.5922, "lr": 3.931851652578137e-05, "epoch": 0.9523809523809523, "percentage": 18.75, "elapsed_time": "0:05:33", "remaining_time": "0:24:05"}
	{"current_steps": 20, "total_steps": 80, "loss": 0.5513, "lr": 3.774021666356444e-05, "epoch": 1.253968253968254, "percentage": 25.0, "elapsed_time": "0:07:13", "remaining_time": "0:21:40"}
	{"current_steps": 25, "total_steps": 80, "loss": 0.5268, "lr": 3.532088886237956e-05, "epoch": 1.5714285714285714, "percentage": 31.25, "elapsed_time": "0:09:03", "remaining_time": "0:19:56"}
	{"current_steps": 30, "total_steps": 80, "loss": 0.5194, "lr": 3.217522858017442e-05, "epoch": 1.8888888888888888, "percentage": 37.5, "elapsed_time": "0:10:56", "remaining_time": "0:18:14"}
	{"current_steps": 35, "total_steps": 80, "loss": 0.4999, "lr": 2.8452365234813992e-05, "epoch": 2.1904761904761907, "percentage": 43.75, "elapsed_time": "0:12:37", "remaining_time": "0:16:14"}
	{"current_steps": 40, "total_steps": 80, "loss": 0.476, "lr": 2.4328792278762058e-05, "epoch": 2.507936507936508, "percentage": 50.0, "elapsed_time": "0:14:24", "remaining_time": "0:14:24"}
	{"current_steps": 45, "total_steps": 80, "loss": 0.4743, "lr": 2e-05, "epoch": 2.825396825396825, "percentage": 56.25, "elapsed_time": "0:16:14", "remaining_time": "0:12:37"}
	{"current_steps": 50, "total_steps": 80, "loss": 0.4803, "lr": 1.5671207721237945e-05, "epoch": 3.126984126984127, "percentage": 62.5, "elapsed_time": "0:17:57", "remaining_time": "0:10:46"}
	{"current_steps": 55, "total_steps": 80, "loss": 0.4741, "lr": 1.1547634765186016e-05, "epoch": 3.4444444444444446, "percentage": 68.75, "elapsed_time": "0:19:43", "remaining_time": "0:08:57"}
	{"current_steps": 60, "total_steps": 80, "loss": 0.4612, "lr": 7.824771419825588e-06, "epoch": 3.761904761904762, "percentage": 75.0, "elapsed_time": "0:21:31", "remaining_time": "0:07:10"}
	{"current_steps": 65, "total_steps": 80, "loss": 0.4606, "lr": 4.679111137620442e-06, "epoch": 4.063492063492063, "percentage": 81.25, "elapsed_time": "0:23:19", "remaining_time": "0:05:22"}
	{"current_steps": 70, "total_steps": 80, "loss": 0.467, "lr": 2.259783336435566e-06, "epoch": 4.380952380952381, "percentage": 87.5, "elapsed_time": "0:25:07", "remaining_time": "0:03:35"}
	{"current_steps": 75, "total_steps": 80, "loss": 0.459, "lr": 6.814834742186361e-07, "epoch": 4.698412698412699, "percentage": 93.75, "elapsed_time": "0:27:01", "remaining_time": "0:01:48"}
	{"current_steps": 80, "total_steps": 80, "loss": 0.4442, "lr": 1.9035568362844037e-08, "epoch": 5.0, "percentage": 100.0, "elapsed_time": "0:28:36", "remaining_time": "0:00:00"}
	{"current_steps": 80, "total_steps": 80, "epoch": 5.0, "percentage": 100.0, "elapsed_time": "0:28:36", "remaining_time": "0:00:00"}