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 146000, 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:b48962ccd06eb3192807ae180f3dd6456f58a152aca65272f73edef4ffb683b1
|
| 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:b3d5eadedde8bbebfeb26d921d794383a8385f17f0c6cc8fb0d00c68ea258e51
|
| 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:51ce9c28a9aa67295d292ad1d8dd89b97ebe967bdf87ecd07447ea36283dbcee
|
| 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:72d1a00336735d13d964c93acea54e818b3d9e40f5ee9551f0bf68590d438f6f
|
| 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,
|
|
@@ -25818,11 +25818,189 @@
|
|
| 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":
|
| 25826 |
"num_train_epochs": 5,
|
| 25827 |
"save_steps": 1000,
|
| 25828 |
"stateful_callbacks": {
|
|
@@ -25837,7 +26015,7 @@
|
|
| 25837 |
"attributes": {}
|
| 25838 |
}
|
| 25839 |
},
|
| 25840 |
-
"total_flos": 1.
|
| 25841 |
"train_batch_size": 32,
|
| 25842 |
"trial_name": null,
|
| 25843 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.207095997852818,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 146000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 25818 |
"eval_steps_per_second": 15.535,
|
| 25819 |
"num_input_tokens_seen": 78631193376,
|
| 25820 |
"step": 145000
|
| 25821 |
+
},
|
| 25822 |
+
{
|
| 25823 |
+
"epoch": 3.186228242447623,
|
| 25824 |
+
"grad_norm": 0.09815175086259842,
|
| 25825 |
+
"learning_rate": 0.0001,
|
| 25826 |
+
"loss": 2.3512,
|
| 25827 |
+
"num_input_tokens_seen": 78683622176,
|
| 25828 |
+
"step": 145050
|
| 25829 |
+
},
|
| 25830 |
+
{
|
| 25831 |
+
"epoch": 3.187326545363686,
|
| 25832 |
+
"grad_norm": 0.08802150189876556,
|
| 25833 |
+
"learning_rate": 0.0001,
|
| 25834 |
+
"loss": 2.352,
|
| 25835 |
+
"num_input_tokens_seen": 78736048064,
|
| 25836 |
+
"step": 145100
|
| 25837 |
+
},
|
| 25838 |
+
{
|
| 25839 |
+
"epoch": 3.1884248482797486,
|
| 25840 |
+
"grad_norm": 0.09322452545166016,
|
| 25841 |
+
"learning_rate": 0.0001,
|
| 25842 |
+
"loss": 2.3517,
|
| 25843 |
+
"num_input_tokens_seen": 78788476864,
|
| 25844 |
+
"step": 145150
|
| 25845 |
+
},
|
| 25846 |
+
{
|
| 25847 |
+
"epoch": 3.1895231511958118,
|
| 25848 |
+
"grad_norm": 0.09273571521043777,
|
| 25849 |
+
"learning_rate": 0.0001,
|
| 25850 |
+
"loss": 2.3525,
|
| 25851 |
+
"num_input_tokens_seen": 78840905664,
|
| 25852 |
+
"step": 145200
|
| 25853 |
+
},
|
| 25854 |
+
{
|
| 25855 |
+
"epoch": 3.1906214541118745,
|
| 25856 |
+
"grad_norm": 0.09510285407304764,
|
| 25857 |
+
"learning_rate": 0.0001,
|
| 25858 |
+
"loss": 2.3492,
|
| 25859 |
+
"num_input_tokens_seen": 78893334464,
|
| 25860 |
+
"step": 145250
|
| 25861 |
+
},
|
| 25862 |
+
{
|
| 25863 |
+
"epoch": 3.191719757027937,
|
| 25864 |
+
"grad_norm": 0.09299364686012268,
|
| 25865 |
+
"learning_rate": 0.0001,
|
| 25866 |
+
"loss": 2.3485,
|
| 25867 |
+
"num_input_tokens_seen": 78945760672,
|
| 25868 |
+
"step": 145300
|
| 25869 |
+
},
|
| 25870 |
+
{
|
| 25871 |
+
"epoch": 3.1928180599440004,
|
| 25872 |
+
"grad_norm": 0.08893129974603653,
|
| 25873 |
+
"learning_rate": 0.0001,
|
| 25874 |
+
"loss": 2.3529,
|
| 25875 |
+
"num_input_tokens_seen": 78998188864,
|
| 25876 |
+
"step": 145350
|
| 25877 |
+
},
|
| 25878 |
+
{
|
| 25879 |
+
"epoch": 3.193916362860063,
|
| 25880 |
+
"grad_norm": 0.08949380367994308,
|
| 25881 |
+
"learning_rate": 0.0001,
|
| 25882 |
+
"loss": 2.3563,
|
| 25883 |
+
"num_input_tokens_seen": 79050616800,
|
| 25884 |
+
"step": 145400
|
| 25885 |
+
},
|
| 25886 |
+
{
|
| 25887 |
+
"epoch": 3.1950146657761263,
|
| 25888 |
+
"grad_norm": 0.09386907517910004,
|
| 25889 |
+
"learning_rate": 0.0001,
|
| 25890 |
+
"loss": 2.3454,
|
| 25891 |
+
"num_input_tokens_seen": 79103043296,
|
| 25892 |
+
"step": 145450
|
| 25893 |
+
},
|
| 25894 |
+
{
|
| 25895 |
+
"epoch": 3.196112968692189,
|
| 25896 |
+
"grad_norm": 0.10135660320520401,
|
| 25897 |
+
"learning_rate": 0.0001,
|
| 25898 |
+
"loss": 2.3521,
|
| 25899 |
+
"num_input_tokens_seen": 79155472096,
|
| 25900 |
+
"step": 145500
|
| 25901 |
+
},
|
| 25902 |
+
{
|
| 25903 |
+
"epoch": 3.196112968692189,
|
| 25904 |
+
"eval_loss": 2.2653002738952637,
|
| 25905 |
+
"eval_runtime": 80.3673,
|
| 25906 |
+
"eval_samples_per_second": 62.214,
|
| 25907 |
+
"eval_steps_per_second": 15.554,
|
| 25908 |
+
"num_input_tokens_seen": 79155472096,
|
| 25909 |
+
"step": 145500
|
| 25910 |
+
},
|
| 25911 |
+
{
|
| 25912 |
+
"epoch": 3.1972112716082517,
|
| 25913 |
+
"grad_norm": 0.09017980098724365,
|
| 25914 |
+
"learning_rate": 0.0001,
|
| 25915 |
+
"loss": 2.3576,
|
| 25916 |
+
"num_input_tokens_seen": 79207895552,
|
| 25917 |
+
"step": 145550
|
| 25918 |
+
},
|
| 25919 |
+
{
|
| 25920 |
+
"epoch": 3.198309574524315,
|
| 25921 |
+
"grad_norm": 0.10214512050151825,
|
| 25922 |
+
"learning_rate": 0.0001,
|
| 25923 |
+
"loss": 2.3512,
|
| 25924 |
+
"num_input_tokens_seen": 79260320832,
|
| 25925 |
+
"step": 145600
|
| 25926 |
+
},
|
| 25927 |
+
{
|
| 25928 |
+
"epoch": 3.1994078774403776,
|
| 25929 |
+
"grad_norm": 0.08758047223091125,
|
| 25930 |
+
"learning_rate": 0.0001,
|
| 25931 |
+
"loss": 2.347,
|
| 25932 |
+
"num_input_tokens_seen": 79312749152,
|
| 25933 |
+
"step": 145650
|
| 25934 |
+
},
|
| 25935 |
+
{
|
| 25936 |
+
"epoch": 3.2005061803564403,
|
| 25937 |
+
"grad_norm": 0.09964236617088318,
|
| 25938 |
+
"learning_rate": 0.0001,
|
| 25939 |
+
"loss": 2.3466,
|
| 25940 |
+
"num_input_tokens_seen": 79365177440,
|
| 25941 |
+
"step": 145700
|
| 25942 |
+
},
|
| 25943 |
+
{
|
| 25944 |
+
"epoch": 3.2016044832725035,
|
| 25945 |
+
"grad_norm": 0.10335622727870941,
|
| 25946 |
+
"learning_rate": 0.0001,
|
| 25947 |
+
"loss": 2.3492,
|
| 25948 |
+
"num_input_tokens_seen": 79417604928,
|
| 25949 |
+
"step": 145750
|
| 25950 |
+
},
|
| 25951 |
+
{
|
| 25952 |
+
"epoch": 3.202702786188566,
|
| 25953 |
+
"grad_norm": 0.09493719041347504,
|
| 25954 |
+
"learning_rate": 0.0001,
|
| 25955 |
+
"loss": 2.3547,
|
| 25956 |
+
"num_input_tokens_seen": 79470031232,
|
| 25957 |
+
"step": 145800
|
| 25958 |
+
},
|
| 25959 |
+
{
|
| 25960 |
+
"epoch": 3.203801089104629,
|
| 25961 |
+
"grad_norm": 0.0902167409658432,
|
| 25962 |
+
"learning_rate": 0.0001,
|
| 25963 |
+
"loss": 2.3449,
|
| 25964 |
+
"num_input_tokens_seen": 79522455456,
|
| 25965 |
+
"step": 145850
|
| 25966 |
+
},
|
| 25967 |
+
{
|
| 25968 |
+
"epoch": 3.204899392020692,
|
| 25969 |
+
"grad_norm": 0.09135492146015167,
|
| 25970 |
+
"learning_rate": 0.0001,
|
| 25971 |
+
"loss": 2.3535,
|
| 25972 |
+
"num_input_tokens_seen": 79574884256,
|
| 25973 |
+
"step": 145900
|
| 25974 |
+
},
|
| 25975 |
+
{
|
| 25976 |
+
"epoch": 3.205997694936755,
|
| 25977 |
+
"grad_norm": 0.09141404926776886,
|
| 25978 |
+
"learning_rate": 0.0001,
|
| 25979 |
+
"loss": 2.3482,
|
| 25980 |
+
"num_input_tokens_seen": 79627304576,
|
| 25981 |
+
"step": 145950
|
| 25982 |
+
},
|
| 25983 |
+
{
|
| 25984 |
+
"epoch": 3.207095997852818,
|
| 25985 |
+
"grad_norm": 0.08859889209270477,
|
| 25986 |
+
"learning_rate": 0.0001,
|
| 25987 |
+
"loss": 2.3585,
|
| 25988 |
+
"num_input_tokens_seen": 79679733376,
|
| 25989 |
+
"step": 146000
|
| 25990 |
+
},
|
| 25991 |
+
{
|
| 25992 |
+
"epoch": 3.207095997852818,
|
| 25993 |
+
"eval_loss": 2.264711618423462,
|
| 25994 |
+
"eval_runtime": 80.4702,
|
| 25995 |
+
"eval_samples_per_second": 62.135,
|
| 25996 |
+
"eval_steps_per_second": 15.534,
|
| 25997 |
+
"num_input_tokens_seen": 79679733376,
|
| 25998 |
+
"step": 146000
|
| 25999 |
}
|
| 26000 |
],
|
| 26001 |
"logging_steps": 50,
|
| 26002 |
"max_steps": 200000,
|
| 26003 |
+
"num_input_tokens_seen": 79679733376,
|
| 26004 |
"num_train_epochs": 5,
|
| 26005 |
"save_steps": 1000,
|
| 26006 |
"stateful_callbacks": {
|
|
|
|
| 26015 |
"attributes": {}
|
| 26016 |
}
|
| 26017 |
},
|
| 26018 |
+
"total_flos": 1.4101859449215222e+20,
|
| 26019 |
"train_batch_size": 32,
|
| 26020 |
"trial_name": null,
|
| 26021 |
"trial_params": null
|