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 152000, 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:d6742b674b0d53e2c3daed10497d7535a63d9e9f7f594d41925fa37702bc1094
|
| 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:eab29f7de023579ff77f6510ad0f7600d1dadaeaa609fc84656241739c587bc8
|
| 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:10363df25df4a89f891174d636e331bcdc2893b7259219b00bec7140fc37ee6b
|
| 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:4ae8b4289a05289ec5bdbf9ddf513572ea179aa9b7d8e7afc52b70d2964dfe0f
|
| 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,
|
|
@@ -26886,11 +26886,189 @@
|
|
| 26886 |
"eval_steps_per_second": 15.519,
|
| 26887 |
"num_input_tokens_seen": 84922496000,
|
| 26888 |
"step": 151000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26889 |
}
|
| 26890 |
],
|
| 26891 |
"logging_steps": 50,
|
| 26892 |
"max_steps": 200000,
|
| 26893 |
-
"num_input_tokens_seen":
|
| 26894 |
"num_train_epochs": 5,
|
| 26895 |
"save_steps": 1000,
|
| 26896 |
"stateful_callbacks": {
|
|
@@ -26905,7 +27083,7 @@
|
|
| 26905 |
"attributes": {}
|
| 26906 |
}
|
| 26907 |
},
|
| 26908 |
-
"total_flos": 1.
|
| 26909 |
"train_batch_size": 32,
|
| 26910 |
"trial_name": null,
|
| 26911 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.338892347780364,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 152000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 26886 |
"eval_steps_per_second": 15.519,
|
| 26887 |
"num_input_tokens_seen": 84922496000,
|
| 26888 |
"step": 151000
|
| 26889 |
+
},
|
| 26890 |
+
{
|
| 26891 |
+
"epoch": 3.3180245923751692,
|
| 26892 |
+
"grad_norm": 0.09862314164638519,
|
| 26893 |
+
"learning_rate": 0.0001,
|
| 26894 |
+
"loss": 2.3591,
|
| 26895 |
+
"num_input_tokens_seen": 84974923744,
|
| 26896 |
+
"step": 151050
|
| 26897 |
+
},
|
| 26898 |
+
{
|
| 26899 |
+
"epoch": 3.3191228952912324,
|
| 26900 |
+
"grad_norm": 0.09108031541109085,
|
| 26901 |
+
"learning_rate": 0.0001,
|
| 26902 |
+
"loss": 2.348,
|
| 26903 |
+
"num_input_tokens_seen": 85027352544,
|
| 26904 |
+
"step": 151100
|
| 26905 |
+
},
|
| 26906 |
+
{
|
| 26907 |
+
"epoch": 3.320221198207295,
|
| 26908 |
+
"grad_norm": 0.10582481324672699,
|
| 26909 |
+
"learning_rate": 0.0001,
|
| 26910 |
+
"loss": 2.3405,
|
| 26911 |
+
"num_input_tokens_seen": 85079779360,
|
| 26912 |
+
"step": 151150
|
| 26913 |
+
},
|
| 26914 |
+
{
|
| 26915 |
+
"epoch": 3.321319501123358,
|
| 26916 |
+
"grad_norm": 0.09384915232658386,
|
| 26917 |
+
"learning_rate": 0.0001,
|
| 26918 |
+
"loss": 2.3497,
|
| 26919 |
+
"num_input_tokens_seen": 85132203008,
|
| 26920 |
+
"step": 151200
|
| 26921 |
+
},
|
| 26922 |
+
{
|
| 26923 |
+
"epoch": 3.322417804039421,
|
| 26924 |
+
"grad_norm": 0.09305764734745026,
|
| 26925 |
+
"learning_rate": 0.0001,
|
| 26926 |
+
"loss": 2.3537,
|
| 26927 |
+
"num_input_tokens_seen": 85184631808,
|
| 26928 |
+
"step": 151250
|
| 26929 |
+
},
|
| 26930 |
+
{
|
| 26931 |
+
"epoch": 3.3235161069554837,
|
| 26932 |
+
"grad_norm": 0.09376391023397446,
|
| 26933 |
+
"learning_rate": 0.0001,
|
| 26934 |
+
"loss": 2.345,
|
| 26935 |
+
"num_input_tokens_seen": 85237060608,
|
| 26936 |
+
"step": 151300
|
| 26937 |
+
},
|
| 26938 |
+
{
|
| 26939 |
+
"epoch": 3.3246144098715464,
|
| 26940 |
+
"grad_norm": 0.09613306075334549,
|
| 26941 |
+
"learning_rate": 0.0001,
|
| 26942 |
+
"loss": 2.3462,
|
| 26943 |
+
"num_input_tokens_seen": 85289487488,
|
| 26944 |
+
"step": 151350
|
| 26945 |
+
},
|
| 26946 |
+
{
|
| 26947 |
+
"epoch": 3.3257127127876096,
|
| 26948 |
+
"grad_norm": 0.09579232335090637,
|
| 26949 |
+
"learning_rate": 0.0001,
|
| 26950 |
+
"loss": 2.3522,
|
| 26951 |
+
"num_input_tokens_seen": 85341909344,
|
| 26952 |
+
"step": 151400
|
| 26953 |
+
},
|
| 26954 |
+
{
|
| 26955 |
+
"epoch": 3.3268110157036723,
|
| 26956 |
+
"grad_norm": 0.08879666030406952,
|
| 26957 |
+
"learning_rate": 0.0001,
|
| 26958 |
+
"loss": 2.353,
|
| 26959 |
+
"num_input_tokens_seen": 85394338144,
|
| 26960 |
+
"step": 151450
|
| 26961 |
+
},
|
| 26962 |
+
{
|
| 26963 |
+
"epoch": 3.327909318619735,
|
| 26964 |
+
"grad_norm": 0.09744927287101746,
|
| 26965 |
+
"learning_rate": 0.0001,
|
| 26966 |
+
"loss": 2.3521,
|
| 26967 |
+
"num_input_tokens_seen": 85446766432,
|
| 26968 |
+
"step": 151500
|
| 26969 |
+
},
|
| 26970 |
+
{
|
| 26971 |
+
"epoch": 3.327909318619735,
|
| 26972 |
+
"eval_loss": 2.26111102104187,
|
| 26973 |
+
"eval_runtime": 79.9827,
|
| 26974 |
+
"eval_samples_per_second": 62.514,
|
| 26975 |
+
"eval_steps_per_second": 15.628,
|
| 26976 |
+
"num_input_tokens_seen": 85446766432,
|
| 26977 |
+
"step": 151500
|
| 26978 |
+
},
|
| 26979 |
+
{
|
| 26980 |
+
"epoch": 3.329007621535798,
|
| 26981 |
+
"grad_norm": 0.0889928862452507,
|
| 26982 |
+
"learning_rate": 0.0001,
|
| 26983 |
+
"loss": 2.3436,
|
| 26984 |
+
"num_input_tokens_seen": 85499195232,
|
| 26985 |
+
"step": 151550
|
| 26986 |
+
},
|
| 26987 |
+
{
|
| 26988 |
+
"epoch": 3.330105924451861,
|
| 26989 |
+
"grad_norm": 0.09036596864461899,
|
| 26990 |
+
"learning_rate": 0.0001,
|
| 26991 |
+
"loss": 2.3525,
|
| 26992 |
+
"num_input_tokens_seen": 85551624032,
|
| 26993 |
+
"step": 151600
|
| 26994 |
+
},
|
| 26995 |
+
{
|
| 26996 |
+
"epoch": 3.331204227367924,
|
| 26997 |
+
"grad_norm": 0.09018707275390625,
|
| 26998 |
+
"learning_rate": 0.0001,
|
| 26999 |
+
"loss": 2.3397,
|
| 27000 |
+
"num_input_tokens_seen": 85604051680,
|
| 27001 |
+
"step": 151650
|
| 27002 |
+
},
|
| 27003 |
+
{
|
| 27004 |
+
"epoch": 3.332302530283987,
|
| 27005 |
+
"grad_norm": 0.09039066731929779,
|
| 27006 |
+
"learning_rate": 0.0001,
|
| 27007 |
+
"loss": 2.3529,
|
| 27008 |
+
"num_input_tokens_seen": 85656480480,
|
| 27009 |
+
"step": 151700
|
| 27010 |
+
},
|
| 27011 |
+
{
|
| 27012 |
+
"epoch": 3.3334008332000495,
|
| 27013 |
+
"grad_norm": 0.09161881357431412,
|
| 27014 |
+
"learning_rate": 0.0001,
|
| 27015 |
+
"loss": 2.3521,
|
| 27016 |
+
"num_input_tokens_seen": 85708908800,
|
| 27017 |
+
"step": 151750
|
| 27018 |
+
},
|
| 27019 |
+
{
|
| 27020 |
+
"epoch": 3.3344991361161127,
|
| 27021 |
+
"grad_norm": 0.09359107166528702,
|
| 27022 |
+
"learning_rate": 0.0001,
|
| 27023 |
+
"loss": 2.3535,
|
| 27024 |
+
"num_input_tokens_seen": 85761337600,
|
| 27025 |
+
"step": 151800
|
| 27026 |
+
},
|
| 27027 |
+
{
|
| 27028 |
+
"epoch": 3.3355974390321754,
|
| 27029 |
+
"grad_norm": 0.09596438705921173,
|
| 27030 |
+
"learning_rate": 0.0001,
|
| 27031 |
+
"loss": 2.3507,
|
| 27032 |
+
"num_input_tokens_seen": 85813766400,
|
| 27033 |
+
"step": 151850
|
| 27034 |
+
},
|
| 27035 |
+
{
|
| 27036 |
+
"epoch": 3.3366957419482386,
|
| 27037 |
+
"grad_norm": 0.09060530364513397,
|
| 27038 |
+
"learning_rate": 0.0001,
|
| 27039 |
+
"loss": 2.3488,
|
| 27040 |
+
"num_input_tokens_seen": 85866195200,
|
| 27041 |
+
"step": 151900
|
| 27042 |
+
},
|
| 27043 |
+
{
|
| 27044 |
+
"epoch": 3.3377940448643013,
|
| 27045 |
+
"grad_norm": 0.09371770173311234,
|
| 27046 |
+
"learning_rate": 0.0001,
|
| 27047 |
+
"loss": 2.3452,
|
| 27048 |
+
"num_input_tokens_seen": 85918618144,
|
| 27049 |
+
"step": 151950
|
| 27050 |
+
},
|
| 27051 |
+
{
|
| 27052 |
+
"epoch": 3.338892347780364,
|
| 27053 |
+
"grad_norm": 0.10104481130838394,
|
| 27054 |
+
"learning_rate": 0.0001,
|
| 27055 |
+
"loss": 2.3474,
|
| 27056 |
+
"num_input_tokens_seen": 85971046944,
|
| 27057 |
+
"step": 152000
|
| 27058 |
+
},
|
| 27059 |
+
{
|
| 27060 |
+
"epoch": 3.338892347780364,
|
| 27061 |
+
"eval_loss": 2.2608485221862793,
|
| 27062 |
+
"eval_runtime": 80.4976,
|
| 27063 |
+
"eval_samples_per_second": 62.114,
|
| 27064 |
+
"eval_steps_per_second": 15.528,
|
| 27065 |
+
"num_input_tokens_seen": 85971046944,
|
| 27066 |
+
"step": 152000
|
| 27067 |
}
|
| 27068 |
],
|
| 27069 |
"logging_steps": 50,
|
| 27070 |
"max_steps": 200000,
|
| 27071 |
+
"num_input_tokens_seen": 85971046944,
|
| 27072 |
"num_train_epochs": 5,
|
| 27073 |
"save_steps": 1000,
|
| 27074 |
"stateful_callbacks": {
|
|
|
|
| 27083 |
"attributes": {}
|
| 27084 |
}
|
| 27085 |
},
|
| 27086 |
+
"total_flos": 1.5215307197191743e+20,
|
| 27087 |
"train_batch_size": 32,
|
| 27088 |
"trial_name": null,
|
| 27089 |
"trial_params": null
|