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 111000, 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:3d918bb0a42fca6cacae9d1bdec880dc5f252219ba38da64e3a58ac700d0c148
|
| 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:a33a3c319d6a66444c7bf9a0b64adb07320f60c802158c31380545be06ce88b0
|
| 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:d1a86d4ecac3fd0dba06cbaf60a1aa4cb43bd245688d3d1e54473ba75e6bfff0
|
| 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:0e6fc7d7e8ba590133db77bb041ad1735fff810d2dc746534b5d880b6609a01e
|
| 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": 1.
|
| 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,
|
|
@@ -19588,11 +19588,189 @@
|
|
| 19588 |
"eval_steps_per_second": 15.003,
|
| 19589 |
"num_input_tokens_seen": 57662439520,
|
| 19590 |
"step": 110000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19591 |
}
|
| 19592 |
],
|
| 19593 |
"logging_steps": 50,
|
| 19594 |
"max_steps": 140000,
|
| 19595 |
-
"num_input_tokens_seen":
|
| 19596 |
"num_train_epochs": 2,
|
| 19597 |
"save_steps": 1000,
|
| 19598 |
"stateful_callbacks": {
|
|
@@ -19607,7 +19785,7 @@
|
|
| 19607 |
"attributes": {}
|
| 19608 |
}
|
| 19609 |
},
|
| 19610 |
-
"total_flos": 1.
|
| 19611 |
"train_batch_size": 32,
|
| 19612 |
"trial_name": null,
|
| 19613 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 1.0589479709505467,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 111000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 19588 |
"eval_steps_per_second": 15.003,
|
| 19589 |
"num_input_tokens_seen": 57662439520,
|
| 19590 |
"step": 110000
|
| 19591 |
+
},
|
| 19592 |
+
{
|
| 19593 |
+
"epoch": 1.0498849231429697,
|
| 19594 |
+
"grad_norm": 0.14630495011806488,
|
| 19595 |
+
"learning_rate": 0.001,
|
| 19596 |
+
"loss": 2.104,
|
| 19597 |
+
"num_input_tokens_seen": 57688651552,
|
| 19598 |
+
"step": 110050
|
| 19599 |
+
},
|
| 19600 |
+
{
|
| 19601 |
+
"epoch": 1.0503619256591579,
|
| 19602 |
+
"grad_norm": 0.15211914479732513,
|
| 19603 |
+
"learning_rate": 0.001,
|
| 19604 |
+
"loss": 2.0929,
|
| 19605 |
+
"num_input_tokens_seen": 57714851968,
|
| 19606 |
+
"step": 110100
|
| 19607 |
+
},
|
| 19608 |
+
{
|
| 19609 |
+
"epoch": 1.050838928175346,
|
| 19610 |
+
"grad_norm": 0.15288826823234558,
|
| 19611 |
+
"learning_rate": 0.001,
|
| 19612 |
+
"loss": 2.107,
|
| 19613 |
+
"num_input_tokens_seen": 57741066368,
|
| 19614 |
+
"step": 110150
|
| 19615 |
+
},
|
| 19616 |
+
{
|
| 19617 |
+
"epoch": 1.0513159306915343,
|
| 19618 |
+
"grad_norm": 0.18201424181461334,
|
| 19619 |
+
"learning_rate": 0.001,
|
| 19620 |
+
"loss": 2.0831,
|
| 19621 |
+
"num_input_tokens_seen": 57767277728,
|
| 19622 |
+
"step": 110200
|
| 19623 |
+
},
|
| 19624 |
+
{
|
| 19625 |
+
"epoch": 1.0517929332077227,
|
| 19626 |
+
"grad_norm": 0.15570518374443054,
|
| 19627 |
+
"learning_rate": 0.001,
|
| 19628 |
+
"loss": 2.1017,
|
| 19629 |
+
"num_input_tokens_seen": 57793489344,
|
| 19630 |
+
"step": 110250
|
| 19631 |
+
},
|
| 19632 |
+
{
|
| 19633 |
+
"epoch": 1.052269935723911,
|
| 19634 |
+
"grad_norm": 0.15590184926986694,
|
| 19635 |
+
"learning_rate": 0.001,
|
| 19636 |
+
"loss": 2.0851,
|
| 19637 |
+
"num_input_tokens_seen": 57819702752,
|
| 19638 |
+
"step": 110300
|
| 19639 |
+
},
|
| 19640 |
+
{
|
| 19641 |
+
"epoch": 1.0527469382400991,
|
| 19642 |
+
"grad_norm": 0.13825973868370056,
|
| 19643 |
+
"learning_rate": 0.001,
|
| 19644 |
+
"loss": 2.0966,
|
| 19645 |
+
"num_input_tokens_seen": 57845893728,
|
| 19646 |
+
"step": 110350
|
| 19647 |
+
},
|
| 19648 |
+
{
|
| 19649 |
+
"epoch": 1.0532239407562876,
|
| 19650 |
+
"grad_norm": 0.1424342691898346,
|
| 19651 |
+
"learning_rate": 0.001,
|
| 19652 |
+
"loss": 2.0915,
|
| 19653 |
+
"num_input_tokens_seen": 57872104704,
|
| 19654 |
+
"step": 110400
|
| 19655 |
+
},
|
| 19656 |
+
{
|
| 19657 |
+
"epoch": 1.0537009432724758,
|
| 19658 |
+
"grad_norm": 0.14073631167411804,
|
| 19659 |
+
"learning_rate": 0.001,
|
| 19660 |
+
"loss": 2.1029,
|
| 19661 |
+
"num_input_tokens_seen": 57898312928,
|
| 19662 |
+
"step": 110450
|
| 19663 |
+
},
|
| 19664 |
+
{
|
| 19665 |
+
"epoch": 1.054177945788664,
|
| 19666 |
+
"grad_norm": 0.1437380015850067,
|
| 19667 |
+
"learning_rate": 0.001,
|
| 19668 |
+
"loss": 2.0945,
|
| 19669 |
+
"num_input_tokens_seen": 57924525696,
|
| 19670 |
+
"step": 110500
|
| 19671 |
+
},
|
| 19672 |
+
{
|
| 19673 |
+
"epoch": 1.054177945788664,
|
| 19674 |
+
"eval_loss": 2.01132869720459,
|
| 19675 |
+
"eval_runtime": 83.1173,
|
| 19676 |
+
"eval_samples_per_second": 60.156,
|
| 19677 |
+
"eval_steps_per_second": 15.039,
|
| 19678 |
+
"num_input_tokens_seen": 57924525696,
|
| 19679 |
+
"step": 110500
|
| 19680 |
+
},
|
| 19681 |
+
{
|
| 19682 |
+
"epoch": 1.0546549483048524,
|
| 19683 |
+
"grad_norm": 0.14192169904708862,
|
| 19684 |
+
"learning_rate": 0.001,
|
| 19685 |
+
"loss": 2.0819,
|
| 19686 |
+
"num_input_tokens_seen": 57950733824,
|
| 19687 |
+
"step": 110550
|
| 19688 |
+
},
|
| 19689 |
+
{
|
| 19690 |
+
"epoch": 1.0551319508210406,
|
| 19691 |
+
"grad_norm": 0.13621965050697327,
|
| 19692 |
+
"learning_rate": 0.001,
|
| 19693 |
+
"loss": 2.096,
|
| 19694 |
+
"num_input_tokens_seen": 57976940256,
|
| 19695 |
+
"step": 110600
|
| 19696 |
+
},
|
| 19697 |
+
{
|
| 19698 |
+
"epoch": 1.0556089533372288,
|
| 19699 |
+
"grad_norm": 0.14394508302211761,
|
| 19700 |
+
"learning_rate": 0.001,
|
| 19701 |
+
"loss": 2.0959,
|
| 19702 |
+
"num_input_tokens_seen": 58003150272,
|
| 19703 |
+
"step": 110650
|
| 19704 |
+
},
|
| 19705 |
+
{
|
| 19706 |
+
"epoch": 1.056085955853417,
|
| 19707 |
+
"grad_norm": 0.15062682330608368,
|
| 19708 |
+
"learning_rate": 0.001,
|
| 19709 |
+
"loss": 2.0912,
|
| 19710 |
+
"num_input_tokens_seen": 58029359264,
|
| 19711 |
+
"step": 110700
|
| 19712 |
+
},
|
| 19713 |
+
{
|
| 19714 |
+
"epoch": 1.0565629583696055,
|
| 19715 |
+
"grad_norm": 0.15145541727542877,
|
| 19716 |
+
"learning_rate": 0.001,
|
| 19717 |
+
"loss": 2.0954,
|
| 19718 |
+
"num_input_tokens_seen": 58055566272,
|
| 19719 |
+
"step": 110750
|
| 19720 |
+
},
|
| 19721 |
+
{
|
| 19722 |
+
"epoch": 1.0570399608857937,
|
| 19723 |
+
"grad_norm": 0.14316266775131226,
|
| 19724 |
+
"learning_rate": 0.001,
|
| 19725 |
+
"loss": 2.0871,
|
| 19726 |
+
"num_input_tokens_seen": 58081774848,
|
| 19727 |
+
"step": 110800
|
| 19728 |
+
},
|
| 19729 |
+
{
|
| 19730 |
+
"epoch": 1.0575169634019819,
|
| 19731 |
+
"grad_norm": 0.1519429087638855,
|
| 19732 |
+
"learning_rate": 0.001,
|
| 19733 |
+
"loss": 2.1033,
|
| 19734 |
+
"num_input_tokens_seen": 58107987648,
|
| 19735 |
+
"step": 110850
|
| 19736 |
+
},
|
| 19737 |
+
{
|
| 19738 |
+
"epoch": 1.0579939659181703,
|
| 19739 |
+
"grad_norm": 0.1556522697210312,
|
| 19740 |
+
"learning_rate": 0.001,
|
| 19741 |
+
"loss": 2.0921,
|
| 19742 |
+
"num_input_tokens_seen": 58134199072,
|
| 19743 |
+
"step": 110900
|
| 19744 |
+
},
|
| 19745 |
+
{
|
| 19746 |
+
"epoch": 1.0584709684343585,
|
| 19747 |
+
"grad_norm": 0.14416266977787018,
|
| 19748 |
+
"learning_rate": 0.001,
|
| 19749 |
+
"loss": 2.0952,
|
| 19750 |
+
"num_input_tokens_seen": 58160410880,
|
| 19751 |
+
"step": 110950
|
| 19752 |
+
},
|
| 19753 |
+
{
|
| 19754 |
+
"epoch": 1.0589479709505467,
|
| 19755 |
+
"grad_norm": 0.14932425320148468,
|
| 19756 |
+
"learning_rate": 0.001,
|
| 19757 |
+
"loss": 2.0978,
|
| 19758 |
+
"num_input_tokens_seen": 58186625280,
|
| 19759 |
+
"step": 111000
|
| 19760 |
+
},
|
| 19761 |
+
{
|
| 19762 |
+
"epoch": 1.0589479709505467,
|
| 19763 |
+
"eval_loss": 2.010718584060669,
|
| 19764 |
+
"eval_runtime": 83.9004,
|
| 19765 |
+
"eval_samples_per_second": 59.594,
|
| 19766 |
+
"eval_steps_per_second": 14.899,
|
| 19767 |
+
"num_input_tokens_seen": 58186625280,
|
| 19768 |
+
"step": 111000
|
| 19769 |
}
|
| 19770 |
],
|
| 19771 |
"logging_steps": 50,
|
| 19772 |
"max_steps": 140000,
|
| 19773 |
+
"num_input_tokens_seen": 58186625280,
|
| 19774 |
"num_train_epochs": 2,
|
| 19775 |
"save_steps": 1000,
|
| 19776 |
"stateful_callbacks": {
|
|
|
|
| 19785 |
"attributes": {}
|
| 19786 |
}
|
| 19787 |
},
|
| 19788 |
+
"total_flos": 1.0297971350514893e+20,
|
| 19789 |
"train_batch_size": 32,
|
| 19790 |
"trial_name": null,
|
| 19791 |
"trial_params": null
|