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 145000, 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:dcc52c805127340a99efe5ce4634aa8da5f6e99e5d84b192bcdfc15db598aaa1
|
| 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:812ced3f153ff459210c79eeaaaf6df9526ae07f07422480db83a62f6a3959ea
|
| 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:e715b06ff1678ef9784950709f202e66ee53f0a61201e7828fbeb571b860f735
|
| 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:9f13aa126d87289d9b7a69890efa6e557270dd33d7688495b067f1fc28a57da6
|
| 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,
|
|
@@ -25640,11 +25640,189 @@
|
|
| 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":
|
| 25648 |
"num_train_epochs": 5,
|
| 25649 |
"save_steps": 1000,
|
| 25650 |
"stateful_callbacks": {
|
|
@@ -25659,7 +25837,7 @@
|
|
| 25659 |
"attributes": {}
|
| 25660 |
}
|
| 25661 |
},
|
| 25662 |
-
"total_flos": 1.
|
| 25663 |
"train_batch_size": 32,
|
| 25664 |
"trial_name": null,
|
| 25665 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.18512993953156,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 145000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 25640 |
"eval_steps_per_second": 15.602,
|
| 25641 |
"num_input_tokens_seen": 77582650016,
|
| 25642 |
"step": 144000
|
| 25643 |
+
},
|
| 25644 |
+
{
|
| 25645 |
+
"epoch": 3.164262184126365,
|
| 25646 |
+
"grad_norm": 0.11045259982347488,
|
| 25647 |
+
"learning_rate": 0.0001,
|
| 25648 |
+
"loss": 2.3539,
|
| 25649 |
+
"num_input_tokens_seen": 77635078816,
|
| 25650 |
+
"step": 144050
|
| 25651 |
+
},
|
| 25652 |
+
{
|
| 25653 |
+
"epoch": 3.1653604870424283,
|
| 25654 |
+
"grad_norm": 0.09217249602079391,
|
| 25655 |
+
"learning_rate": 0.0001,
|
| 25656 |
+
"loss": 2.343,
|
| 25657 |
+
"num_input_tokens_seen": 77687507616,
|
| 25658 |
+
"step": 144100
|
| 25659 |
+
},
|
| 25660 |
+
{
|
| 25661 |
+
"epoch": 3.166458789958491,
|
| 25662 |
+
"grad_norm": 0.09735982865095139,
|
| 25663 |
+
"learning_rate": 0.0001,
|
| 25664 |
+
"loss": 2.3519,
|
| 25665 |
+
"num_input_tokens_seen": 77739935872,
|
| 25666 |
+
"step": 144150
|
| 25667 |
+
},
|
| 25668 |
+
{
|
| 25669 |
+
"epoch": 3.167557092874554,
|
| 25670 |
+
"grad_norm": 0.08962323516607285,
|
| 25671 |
+
"learning_rate": 0.0001,
|
| 25672 |
+
"loss": 2.3473,
|
| 25673 |
+
"num_input_tokens_seen": 77792364672,
|
| 25674 |
+
"step": 144200
|
| 25675 |
+
},
|
| 25676 |
+
{
|
| 25677 |
+
"epoch": 3.168655395790617,
|
| 25678 |
+
"grad_norm": 0.09209229052066803,
|
| 25679 |
+
"learning_rate": 0.0001,
|
| 25680 |
+
"loss": 2.3555,
|
| 25681 |
+
"num_input_tokens_seen": 77844789440,
|
| 25682 |
+
"step": 144250
|
| 25683 |
+
},
|
| 25684 |
+
{
|
| 25685 |
+
"epoch": 3.1697536987066797,
|
| 25686 |
+
"grad_norm": 0.09181981533765793,
|
| 25687 |
+
"learning_rate": 0.0001,
|
| 25688 |
+
"loss": 2.348,
|
| 25689 |
+
"num_input_tokens_seen": 77897216224,
|
| 25690 |
+
"step": 144300
|
| 25691 |
+
},
|
| 25692 |
+
{
|
| 25693 |
+
"epoch": 3.1708520016227424,
|
| 25694 |
+
"grad_norm": 0.0833621546626091,
|
| 25695 |
+
"learning_rate": 0.0001,
|
| 25696 |
+
"loss": 2.3454,
|
| 25697 |
+
"num_input_tokens_seen": 77949645024,
|
| 25698 |
+
"step": 144350
|
| 25699 |
+
},
|
| 25700 |
+
{
|
| 25701 |
+
"epoch": 3.1719503045388056,
|
| 25702 |
+
"grad_norm": 0.09470181167125702,
|
| 25703 |
+
"learning_rate": 0.0001,
|
| 25704 |
+
"loss": 2.3491,
|
| 25705 |
+
"num_input_tokens_seen": 78002073440,
|
| 25706 |
+
"step": 144400
|
| 25707 |
+
},
|
| 25708 |
+
{
|
| 25709 |
+
"epoch": 3.1730486074548683,
|
| 25710 |
+
"grad_norm": 0.09017711132764816,
|
| 25711 |
+
"learning_rate": 0.0001,
|
| 25712 |
+
"loss": 2.3545,
|
| 25713 |
+
"num_input_tokens_seen": 78054502240,
|
| 25714 |
+
"step": 144450
|
| 25715 |
+
},
|
| 25716 |
+
{
|
| 25717 |
+
"epoch": 3.174146910370931,
|
| 25718 |
+
"grad_norm": 0.09220907092094421,
|
| 25719 |
+
"learning_rate": 0.0001,
|
| 25720 |
+
"loss": 2.3508,
|
| 25721 |
+
"num_input_tokens_seen": 78106920224,
|
| 25722 |
+
"step": 144500
|
| 25723 |
+
},
|
| 25724 |
+
{
|
| 25725 |
+
"epoch": 3.174146910370931,
|
| 25726 |
+
"eval_loss": 2.2662734985351562,
|
| 25727 |
+
"eval_runtime": 80.1838,
|
| 25728 |
+
"eval_samples_per_second": 62.357,
|
| 25729 |
+
"eval_steps_per_second": 15.589,
|
| 25730 |
+
"num_input_tokens_seen": 78106920224,
|
| 25731 |
+
"step": 144500
|
| 25732 |
+
},
|
| 25733 |
+
{
|
| 25734 |
+
"epoch": 3.175245213286994,
|
| 25735 |
+
"grad_norm": 0.0899912640452385,
|
| 25736 |
+
"learning_rate": 0.0001,
|
| 25737 |
+
"loss": 2.3607,
|
| 25738 |
+
"num_input_tokens_seen": 78159346816,
|
| 25739 |
+
"step": 144550
|
| 25740 |
+
},
|
| 25741 |
+
{
|
| 25742 |
+
"epoch": 3.176343516203057,
|
| 25743 |
+
"grad_norm": 0.09542039781808853,
|
| 25744 |
+
"learning_rate": 0.0001,
|
| 25745 |
+
"loss": 2.3405,
|
| 25746 |
+
"num_input_tokens_seen": 78211775616,
|
| 25747 |
+
"step": 144600
|
| 25748 |
+
},
|
| 25749 |
+
{
|
| 25750 |
+
"epoch": 3.17744181911912,
|
| 25751 |
+
"grad_norm": 0.09049233049154282,
|
| 25752 |
+
"learning_rate": 0.0001,
|
| 25753 |
+
"loss": 2.3483,
|
| 25754 |
+
"num_input_tokens_seen": 78264204032,
|
| 25755 |
+
"step": 144650
|
| 25756 |
+
},
|
| 25757 |
+
{
|
| 25758 |
+
"epoch": 3.1785401220351828,
|
| 25759 |
+
"grad_norm": 0.09777910262346268,
|
| 25760 |
+
"learning_rate": 0.0001,
|
| 25761 |
+
"loss": 2.3514,
|
| 25762 |
+
"num_input_tokens_seen": 78316631456,
|
| 25763 |
+
"step": 144700
|
| 25764 |
+
},
|
| 25765 |
+
{
|
| 25766 |
+
"epoch": 3.1796384249512455,
|
| 25767 |
+
"grad_norm": 0.09022051095962524,
|
| 25768 |
+
"learning_rate": 0.0001,
|
| 25769 |
+
"loss": 2.3531,
|
| 25770 |
+
"num_input_tokens_seen": 78369060192,
|
| 25771 |
+
"step": 144750
|
| 25772 |
+
},
|
| 25773 |
+
{
|
| 25774 |
+
"epoch": 3.1807367278673087,
|
| 25775 |
+
"grad_norm": 0.09529806673526764,
|
| 25776 |
+
"learning_rate": 0.0001,
|
| 25777 |
+
"loss": 2.3531,
|
| 25778 |
+
"num_input_tokens_seen": 78421488992,
|
| 25779 |
+
"step": 144800
|
| 25780 |
+
},
|
| 25781 |
+
{
|
| 25782 |
+
"epoch": 3.1818350307833714,
|
| 25783 |
+
"grad_norm": 0.08957728743553162,
|
| 25784 |
+
"learning_rate": 0.0001,
|
| 25785 |
+
"loss": 2.3554,
|
| 25786 |
+
"num_input_tokens_seen": 78473915424,
|
| 25787 |
+
"step": 144850
|
| 25788 |
+
},
|
| 25789 |
+
{
|
| 25790 |
+
"epoch": 3.182933333699434,
|
| 25791 |
+
"grad_norm": 0.09089584648609161,
|
| 25792 |
+
"learning_rate": 0.0001,
|
| 25793 |
+
"loss": 2.34,
|
| 25794 |
+
"num_input_tokens_seen": 78526342656,
|
| 25795 |
+
"step": 144900
|
| 25796 |
+
},
|
| 25797 |
+
{
|
| 25798 |
+
"epoch": 3.1840316366154973,
|
| 25799 |
+
"grad_norm": 0.09180238097906113,
|
| 25800 |
+
"learning_rate": 0.0001,
|
| 25801 |
+
"loss": 2.3542,
|
| 25802 |
+
"num_input_tokens_seen": 78578771456,
|
| 25803 |
+
"step": 144950
|
| 25804 |
+
},
|
| 25805 |
+
{
|
| 25806 |
+
"epoch": 3.18512993953156,
|
| 25807 |
+
"grad_norm": 0.09355945885181427,
|
| 25808 |
+
"learning_rate": 0.0001,
|
| 25809 |
+
"loss": 2.3581,
|
| 25810 |
+
"num_input_tokens_seen": 78631193376,
|
| 25811 |
+
"step": 145000
|
| 25812 |
+
},
|
| 25813 |
+
{
|
| 25814 |
+
"epoch": 3.18512993953156,
|
| 25815 |
+
"eval_loss": 2.2656116485595703,
|
| 25816 |
+
"eval_runtime": 80.4625,
|
| 25817 |
+
"eval_samples_per_second": 62.141,
|
| 25818 |
+
"eval_steps_per_second": 15.535,
|
| 25819 |
+
"num_input_tokens_seen": 78631193376,
|
| 25820 |
+
"step": 145000
|
| 25821 |
}
|
| 25822 |
],
|
| 25823 |
"logging_steps": 50,
|
| 25824 |
"max_steps": 200000,
|
| 25825 |
+
"num_input_tokens_seen": 78631193376,
|
| 25826 |
"num_train_epochs": 5,
|
| 25827 |
"save_steps": 1000,
|
| 25828 |
"stateful_callbacks": {
|
|
|
|
| 25837 |
"attributes": {}
|
| 25838 |
}
|
| 25839 |
},
|
| 25840 |
+
"total_flos": 1.3916286994584822e+20,
|
| 25841 |
"train_batch_size": 32,
|
| 25842 |
"trial_name": null,
|
| 25843 |
"trial_params": null
|