Instructions to use Azrail/smallm_350 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Azrail/smallm_350 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Azrail/smallm_350", trust_remote_code=True)# pip install -U transformers accelerate # Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("Azrail/smallm_350", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use Azrail/smallm_350 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "Azrail/smallm_350" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Azrail/smallm_350", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/Azrail/smallm_350
- SGLang
How to use Azrail/smallm_350 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 "Azrail/smallm_350" \ --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": "Azrail/smallm_350", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'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 "Azrail/smallm_350" \ --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": "Azrail/smallm_350", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use Azrail/smallm_350 with Docker Model Runner:
docker model run hf.co/Azrail/smallm_350
Training in progress, step 144000, checkpoint
Browse files
last-checkpoint/model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1410301944
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0c6c980ace5ce0a4fd3625a02e163df067832f4ff0cce1c1c305ee9d44e17b3b
|
| 3 |
size 1410301944
|
last-checkpoint/optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2820185786
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fd3b39bcee85f2c89b3dd95a7ec3fbeee713f0e76059470f8cf9bb3b9084c3b2
|
| 3 |
size 2820185786
|
last-checkpoint/rng_state.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14244
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d150d74b6234978f1de50d9684bb41dd3c162274dba720cabd22f3c0df64b646
|
| 3 |
size 14244
|
last-checkpoint/scheduler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1064
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:62e501f63b681f89119727a00e92acc6c6fd02c59ff9044b7e19b90a76e351b2
|
| 3 |
size 1064
|
last-checkpoint/trainer_state.json
CHANGED
|
@@ -2,9 +2,9 @@
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
-
"epoch": 3.
|
| 6 |
"eval_steps": 500,
|
| 7 |
-
"global_step":
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
@@ -25462,11 +25462,189 @@
|
|
| 25462 |
"eval_steps_per_second": 15.476,
|
| 25463 |
"num_input_tokens_seen": 76534108256,
|
| 25464 |
"step": 143000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25465 |
}
|
| 25466 |
],
|
| 25467 |
"logging_steps": 50,
|
| 25468 |
"max_steps": 200000,
|
| 25469 |
-
"num_input_tokens_seen":
|
| 25470 |
"num_train_epochs": 5,
|
| 25471 |
"save_steps": 1000,
|
| 25472 |
"stateful_callbacks": {
|
|
@@ -25481,7 +25659,7 @@
|
|
| 25481 |
"attributes": {}
|
| 25482 |
}
|
| 25483 |
},
|
| 25484 |
-
"total_flos": 1.
|
| 25485 |
"train_batch_size": 32,
|
| 25486 |
"trial_name": null,
|
| 25487 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.1631638812103025,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 144000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 25462 |
"eval_steps_per_second": 15.476,
|
| 25463 |
"num_input_tokens_seen": 76534108256,
|
| 25464 |
"step": 143000
|
| 25465 |
+
},
|
| 25466 |
+
{
|
| 25467 |
+
"epoch": 3.1422961258051076,
|
| 25468 |
+
"grad_norm": 0.0887688547372818,
|
| 25469 |
+
"learning_rate": 0.0001,
|
| 25470 |
+
"loss": 2.3614,
|
| 25471 |
+
"num_input_tokens_seen": 76586530944,
|
| 25472 |
+
"step": 143050
|
| 25473 |
+
},
|
| 25474 |
+
{
|
| 25475 |
+
"epoch": 3.1433944287211704,
|
| 25476 |
+
"grad_norm": 0.09098675847053528,
|
| 25477 |
+
"learning_rate": 0.0001,
|
| 25478 |
+
"loss": 2.3579,
|
| 25479 |
+
"num_input_tokens_seen": 76638954624,
|
| 25480 |
+
"step": 143100
|
| 25481 |
+
},
|
| 25482 |
+
{
|
| 25483 |
+
"epoch": 3.144492731637233,
|
| 25484 |
+
"grad_norm": 0.0879465788602829,
|
| 25485 |
+
"learning_rate": 0.0001,
|
| 25486 |
+
"loss": 2.3544,
|
| 25487 |
+
"num_input_tokens_seen": 76691382816,
|
| 25488 |
+
"step": 143150
|
| 25489 |
+
},
|
| 25490 |
+
{
|
| 25491 |
+
"epoch": 3.1455910345532963,
|
| 25492 |
+
"grad_norm": 0.09020204097032547,
|
| 25493 |
+
"learning_rate": 0.0001,
|
| 25494 |
+
"loss": 2.3608,
|
| 25495 |
+
"num_input_tokens_seen": 76743811616,
|
| 25496 |
+
"step": 143200
|
| 25497 |
+
},
|
| 25498 |
+
{
|
| 25499 |
+
"epoch": 3.146689337469359,
|
| 25500 |
+
"grad_norm": 0.0859370082616806,
|
| 25501 |
+
"learning_rate": 0.0001,
|
| 25502 |
+
"loss": 2.3572,
|
| 25503 |
+
"num_input_tokens_seen": 76796240416,
|
| 25504 |
+
"step": 143250
|
| 25505 |
+
},
|
| 25506 |
+
{
|
| 25507 |
+
"epoch": 3.147787640385422,
|
| 25508 |
+
"grad_norm": 0.08659014105796814,
|
| 25509 |
+
"learning_rate": 0.0001,
|
| 25510 |
+
"loss": 2.3513,
|
| 25511 |
+
"num_input_tokens_seen": 76848663776,
|
| 25512 |
+
"step": 143300
|
| 25513 |
+
},
|
| 25514 |
+
{
|
| 25515 |
+
"epoch": 3.148885943301485,
|
| 25516 |
+
"grad_norm": 0.09119068831205368,
|
| 25517 |
+
"learning_rate": 0.0001,
|
| 25518 |
+
"loss": 2.3591,
|
| 25519 |
+
"num_input_tokens_seen": 76901090432,
|
| 25520 |
+
"step": 143350
|
| 25521 |
+
},
|
| 25522 |
+
{
|
| 25523 |
+
"epoch": 3.1499842462175476,
|
| 25524 |
+
"grad_norm": 0.08679833263158798,
|
| 25525 |
+
"learning_rate": 0.0001,
|
| 25526 |
+
"loss": 2.3462,
|
| 25527 |
+
"num_input_tokens_seen": 76953519232,
|
| 25528 |
+
"step": 143400
|
| 25529 |
+
},
|
| 25530 |
+
{
|
| 25531 |
+
"epoch": 3.1510825491336107,
|
| 25532 |
+
"grad_norm": 0.09158121794462204,
|
| 25533 |
+
"learning_rate": 0.0001,
|
| 25534 |
+
"loss": 2.354,
|
| 25535 |
+
"num_input_tokens_seen": 77005948032,
|
| 25536 |
+
"step": 143450
|
| 25537 |
+
},
|
| 25538 |
+
{
|
| 25539 |
+
"epoch": 3.1521808520496735,
|
| 25540 |
+
"grad_norm": 0.08906587958335876,
|
| 25541 |
+
"learning_rate": 0.0001,
|
| 25542 |
+
"loss": 2.3582,
|
| 25543 |
+
"num_input_tokens_seen": 77058376736,
|
| 25544 |
+
"step": 143500
|
| 25545 |
+
},
|
| 25546 |
+
{
|
| 25547 |
+
"epoch": 3.1521808520496735,
|
| 25548 |
+
"eval_loss": 2.2670507431030273,
|
| 25549 |
+
"eval_runtime": 80.7537,
|
| 25550 |
+
"eval_samples_per_second": 61.917,
|
| 25551 |
+
"eval_steps_per_second": 15.479,
|
| 25552 |
+
"num_input_tokens_seen": 77058376736,
|
| 25553 |
+
"step": 143500
|
| 25554 |
+
},
|
| 25555 |
+
{
|
| 25556 |
+
"epoch": 3.153279154965736,
|
| 25557 |
+
"grad_norm": 0.09259413182735443,
|
| 25558 |
+
"learning_rate": 0.0001,
|
| 25559 |
+
"loss": 2.3501,
|
| 25560 |
+
"num_input_tokens_seen": 77110805536,
|
| 25561 |
+
"step": 143550
|
| 25562 |
+
},
|
| 25563 |
+
{
|
| 25564 |
+
"epoch": 3.1543774578817994,
|
| 25565 |
+
"grad_norm": 0.08947575837373734,
|
| 25566 |
+
"learning_rate": 0.0001,
|
| 25567 |
+
"loss": 2.3595,
|
| 25568 |
+
"num_input_tokens_seen": 77163233984,
|
| 25569 |
+
"step": 143600
|
| 25570 |
+
},
|
| 25571 |
+
{
|
| 25572 |
+
"epoch": 3.155475760797862,
|
| 25573 |
+
"grad_norm": 0.09320012480020523,
|
| 25574 |
+
"learning_rate": 0.0001,
|
| 25575 |
+
"loss": 2.3566,
|
| 25576 |
+
"num_input_tokens_seen": 77215662784,
|
| 25577 |
+
"step": 143650
|
| 25578 |
+
},
|
| 25579 |
+
{
|
| 25580 |
+
"epoch": 3.1565740637139252,
|
| 25581 |
+
"grad_norm": 0.09630698710680008,
|
| 25582 |
+
"learning_rate": 0.0001,
|
| 25583 |
+
"loss": 2.3619,
|
| 25584 |
+
"num_input_tokens_seen": 77268081952,
|
| 25585 |
+
"step": 143700
|
| 25586 |
+
},
|
| 25587 |
+
{
|
| 25588 |
+
"epoch": 3.157672366629988,
|
| 25589 |
+
"grad_norm": 0.08778363466262817,
|
| 25590 |
+
"learning_rate": 0.0001,
|
| 25591 |
+
"loss": 2.3565,
|
| 25592 |
+
"num_input_tokens_seen": 77320510752,
|
| 25593 |
+
"step": 143750
|
| 25594 |
+
},
|
| 25595 |
+
{
|
| 25596 |
+
"epoch": 3.1587706695460507,
|
| 25597 |
+
"grad_norm": 0.08947426080703735,
|
| 25598 |
+
"learning_rate": 0.0001,
|
| 25599 |
+
"loss": 2.3497,
|
| 25600 |
+
"num_input_tokens_seen": 77372937824,
|
| 25601 |
+
"step": 143800
|
| 25602 |
+
},
|
| 25603 |
+
{
|
| 25604 |
+
"epoch": 3.159868972462114,
|
| 25605 |
+
"grad_norm": 0.09249094128608704,
|
| 25606 |
+
"learning_rate": 0.0001,
|
| 25607 |
+
"loss": 2.3508,
|
| 25608 |
+
"num_input_tokens_seen": 77425366624,
|
| 25609 |
+
"step": 143850
|
| 25610 |
+
},
|
| 25611 |
+
{
|
| 25612 |
+
"epoch": 3.1609672753781766,
|
| 25613 |
+
"grad_norm": 0.09168736636638641,
|
| 25614 |
+
"learning_rate": 0.0001,
|
| 25615 |
+
"loss": 2.3522,
|
| 25616 |
+
"num_input_tokens_seen": 77477795424,
|
| 25617 |
+
"step": 143900
|
| 25618 |
+
},
|
| 25619 |
+
{
|
| 25620 |
+
"epoch": 3.1620655782942393,
|
| 25621 |
+
"grad_norm": 0.08995141088962555,
|
| 25622 |
+
"learning_rate": 0.0001,
|
| 25623 |
+
"loss": 2.3506,
|
| 25624 |
+
"num_input_tokens_seen": 77530222464,
|
| 25625 |
+
"step": 143950
|
| 25626 |
+
},
|
| 25627 |
+
{
|
| 25628 |
+
"epoch": 3.1631638812103025,
|
| 25629 |
+
"grad_norm": 0.09521841257810593,
|
| 25630 |
+
"learning_rate": 0.0001,
|
| 25631 |
+
"loss": 2.3521,
|
| 25632 |
+
"num_input_tokens_seen": 77582650016,
|
| 25633 |
+
"step": 144000
|
| 25634 |
+
},
|
| 25635 |
+
{
|
| 25636 |
+
"epoch": 3.1631638812103025,
|
| 25637 |
+
"eval_loss": 2.266408920288086,
|
| 25638 |
+
"eval_runtime": 80.1189,
|
| 25639 |
+
"eval_samples_per_second": 62.407,
|
| 25640 |
+
"eval_steps_per_second": 15.602,
|
| 25641 |
+
"num_input_tokens_seen": 77582650016,
|
| 25642 |
+
"step": 144000
|
| 25643 |
}
|
| 25644 |
],
|
| 25645 |
"logging_steps": 50,
|
| 25646 |
"max_steps": 200000,
|
| 25647 |
+
"num_input_tokens_seen": 77582650016,
|
| 25648 |
"num_train_epochs": 5,
|
| 25649 |
"save_steps": 1000,
|
| 25650 |
"stateful_callbacks": {
|
|
|
|
| 25659 |
"attributes": {}
|
| 25660 |
}
|
| 25661 |
},
|
| 25662 |
+
"total_flos": 1.3730713945295708e+20,
|
| 25663 |
"train_batch_size": 32,
|
| 25664 |
"trial_name": null,
|
| 25665 |
"trial_params": null
|