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 153000, 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:49e33340944bdfba019639a505fb829fbf58de481dcbd6bfa34260585e39bc82
|
| 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:ea0c2ad8c2577dec58bcf798facca9b484958ae7510c46df4fbc524b7011e74a
|
| 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:2bb1d6d2c94dbd4fc1bf9d5756f17ab78e6ccfa57b4112f0c257527e9ee1e447
|
| 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:6a693861125ab0b2f85599d0c1f06c209c17c8fe0c9515e61fbc5e32f352f900
|
| 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,
|
|
@@ -27064,11 +27064,189 @@
|
|
| 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":
|
| 27072 |
"num_train_epochs": 5,
|
| 27073 |
"save_steps": 1000,
|
| 27074 |
"stateful_callbacks": {
|
|
@@ -27083,7 +27261,7 @@
|
|
| 27083 |
"attributes": {}
|
| 27084 |
}
|
| 27085 |
},
|
| 27086 |
-
"total_flos": 1.
|
| 27087 |
"train_batch_size": 32,
|
| 27088 |
"trial_name": null,
|
| 27089 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.360858406101622,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 153000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 27064 |
"eval_steps_per_second": 15.528,
|
| 27065 |
"num_input_tokens_seen": 85971046944,
|
| 27066 |
"step": 152000
|
| 27067 |
+
},
|
| 27068 |
+
{
|
| 27069 |
+
"epoch": 3.339990650696427,
|
| 27070 |
+
"grad_norm": 0.10089096426963806,
|
| 27071 |
+
"learning_rate": 0.0001,
|
| 27072 |
+
"loss": 2.3468,
|
| 27073 |
+
"num_input_tokens_seen": 86023475744,
|
| 27074 |
+
"step": 152050
|
| 27075 |
+
},
|
| 27076 |
+
{
|
| 27077 |
+
"epoch": 3.34108895361249,
|
| 27078 |
+
"grad_norm": 0.093196339905262,
|
| 27079 |
+
"learning_rate": 0.0001,
|
| 27080 |
+
"loss": 2.351,
|
| 27081 |
+
"num_input_tokens_seen": 86075903552,
|
| 27082 |
+
"step": 152100
|
| 27083 |
+
},
|
| 27084 |
+
{
|
| 27085 |
+
"epoch": 3.3421872565285526,
|
| 27086 |
+
"grad_norm": 0.09717004001140594,
|
| 27087 |
+
"learning_rate": 0.0001,
|
| 27088 |
+
"loss": 2.3486,
|
| 27089 |
+
"num_input_tokens_seen": 86128331904,
|
| 27090 |
+
"step": 152150
|
| 27091 |
+
},
|
| 27092 |
+
{
|
| 27093 |
+
"epoch": 3.343285559444616,
|
| 27094 |
+
"grad_norm": 0.09837143868207932,
|
| 27095 |
+
"learning_rate": 0.0001,
|
| 27096 |
+
"loss": 2.3424,
|
| 27097 |
+
"num_input_tokens_seen": 86180760704,
|
| 27098 |
+
"step": 152200
|
| 27099 |
+
},
|
| 27100 |
+
{
|
| 27101 |
+
"epoch": 3.3443838623606785,
|
| 27102 |
+
"grad_norm": 0.08969301730394363,
|
| 27103 |
+
"learning_rate": 0.0001,
|
| 27104 |
+
"loss": 2.355,
|
| 27105 |
+
"num_input_tokens_seen": 86233189120,
|
| 27106 |
+
"step": 152250
|
| 27107 |
+
},
|
| 27108 |
+
{
|
| 27109 |
+
"epoch": 3.3454821652767412,
|
| 27110 |
+
"grad_norm": 0.09237448871135712,
|
| 27111 |
+
"learning_rate": 0.0001,
|
| 27112 |
+
"loss": 2.3472,
|
| 27113 |
+
"num_input_tokens_seen": 86285617920,
|
| 27114 |
+
"step": 152300
|
| 27115 |
+
},
|
| 27116 |
+
{
|
| 27117 |
+
"epoch": 3.3465804681928044,
|
| 27118 |
+
"grad_norm": 0.08787325024604797,
|
| 27119 |
+
"learning_rate": 0.0001,
|
| 27120 |
+
"loss": 2.3456,
|
| 27121 |
+
"num_input_tokens_seen": 86338046720,
|
| 27122 |
+
"step": 152350
|
| 27123 |
+
},
|
| 27124 |
+
{
|
| 27125 |
+
"epoch": 3.347678771108867,
|
| 27126 |
+
"grad_norm": 0.0857023298740387,
|
| 27127 |
+
"learning_rate": 0.0001,
|
| 27128 |
+
"loss": 2.3511,
|
| 27129 |
+
"num_input_tokens_seen": 86390472640,
|
| 27130 |
+
"step": 152400
|
| 27131 |
+
},
|
| 27132 |
+
{
|
| 27133 |
+
"epoch": 3.3487770740249303,
|
| 27134 |
+
"grad_norm": 0.09470528364181519,
|
| 27135 |
+
"learning_rate": 0.0001,
|
| 27136 |
+
"loss": 2.3481,
|
| 27137 |
+
"num_input_tokens_seen": 86442901440,
|
| 27138 |
+
"step": 152450
|
| 27139 |
+
},
|
| 27140 |
+
{
|
| 27141 |
+
"epoch": 3.349875376940993,
|
| 27142 |
+
"grad_norm": 0.08560249954462051,
|
| 27143 |
+
"learning_rate": 0.0001,
|
| 27144 |
+
"loss": 2.3419,
|
| 27145 |
+
"num_input_tokens_seen": 86495325760,
|
| 27146 |
+
"step": 152500
|
| 27147 |
+
},
|
| 27148 |
+
{
|
| 27149 |
+
"epoch": 3.349875376940993,
|
| 27150 |
+
"eval_loss": 2.260103225708008,
|
| 27151 |
+
"eval_runtime": 79.9152,
|
| 27152 |
+
"eval_samples_per_second": 62.566,
|
| 27153 |
+
"eval_steps_per_second": 15.642,
|
| 27154 |
+
"num_input_tokens_seen": 86495325760,
|
| 27155 |
+
"step": 152500
|
| 27156 |
+
},
|
| 27157 |
+
{
|
| 27158 |
+
"epoch": 3.3509736798570557,
|
| 27159 |
+
"grad_norm": 0.1033218502998352,
|
| 27160 |
+
"learning_rate": 0.0001,
|
| 27161 |
+
"loss": 2.3473,
|
| 27162 |
+
"num_input_tokens_seen": 86547749728,
|
| 27163 |
+
"step": 152550
|
| 27164 |
+
},
|
| 27165 |
+
{
|
| 27166 |
+
"epoch": 3.352071982773119,
|
| 27167 |
+
"grad_norm": 0.08745966106653214,
|
| 27168 |
+
"learning_rate": 0.0001,
|
| 27169 |
+
"loss": 2.3524,
|
| 27170 |
+
"num_input_tokens_seen": 86600178048,
|
| 27171 |
+
"step": 152600
|
| 27172 |
+
},
|
| 27173 |
+
{
|
| 27174 |
+
"epoch": 3.3531702856891816,
|
| 27175 |
+
"grad_norm": 0.08843880146741867,
|
| 27176 |
+
"learning_rate": 0.0001,
|
| 27177 |
+
"loss": 2.3467,
|
| 27178 |
+
"num_input_tokens_seen": 86652604992,
|
| 27179 |
+
"step": 152650
|
| 27180 |
+
},
|
| 27181 |
+
{
|
| 27182 |
+
"epoch": 3.3542685886052444,
|
| 27183 |
+
"grad_norm": 0.09221280366182327,
|
| 27184 |
+
"learning_rate": 0.0001,
|
| 27185 |
+
"loss": 2.3472,
|
| 27186 |
+
"num_input_tokens_seen": 86705033792,
|
| 27187 |
+
"step": 152700
|
| 27188 |
+
},
|
| 27189 |
+
{
|
| 27190 |
+
"epoch": 3.3553668915213075,
|
| 27191 |
+
"grad_norm": 0.09814682602882385,
|
| 27192 |
+
"learning_rate": 0.0001,
|
| 27193 |
+
"loss": 2.3482,
|
| 27194 |
+
"num_input_tokens_seen": 86757462592,
|
| 27195 |
+
"step": 152750
|
| 27196 |
+
},
|
| 27197 |
+
{
|
| 27198 |
+
"epoch": 3.3564651944373702,
|
| 27199 |
+
"grad_norm": 0.09335148334503174,
|
| 27200 |
+
"learning_rate": 0.0001,
|
| 27201 |
+
"loss": 2.3454,
|
| 27202 |
+
"num_input_tokens_seen": 86809891392,
|
| 27203 |
+
"step": 152800
|
| 27204 |
+
},
|
| 27205 |
+
{
|
| 27206 |
+
"epoch": 3.357563497353433,
|
| 27207 |
+
"grad_norm": 0.09705245494842529,
|
| 27208 |
+
"learning_rate": 0.0001,
|
| 27209 |
+
"loss": 2.351,
|
| 27210 |
+
"num_input_tokens_seen": 86862320192,
|
| 27211 |
+
"step": 152850
|
| 27212 |
+
},
|
| 27213 |
+
{
|
| 27214 |
+
"epoch": 3.358661800269496,
|
| 27215 |
+
"grad_norm": 0.09022082388401031,
|
| 27216 |
+
"learning_rate": 0.0001,
|
| 27217 |
+
"loss": 2.3474,
|
| 27218 |
+
"num_input_tokens_seen": 86914748672,
|
| 27219 |
+
"step": 152900
|
| 27220 |
+
},
|
| 27221 |
+
{
|
| 27222 |
+
"epoch": 3.359760103185559,
|
| 27223 |
+
"grad_norm": 0.09682248532772064,
|
| 27224 |
+
"learning_rate": 0.0001,
|
| 27225 |
+
"loss": 2.3424,
|
| 27226 |
+
"num_input_tokens_seen": 86967177472,
|
| 27227 |
+
"step": 152950
|
| 27228 |
+
},
|
| 27229 |
+
{
|
| 27230 |
+
"epoch": 3.360858406101622,
|
| 27231 |
+
"grad_norm": 0.092010997235775,
|
| 27232 |
+
"learning_rate": 0.0001,
|
| 27233 |
+
"loss": 2.3463,
|
| 27234 |
+
"num_input_tokens_seen": 87019603744,
|
| 27235 |
+
"step": 153000
|
| 27236 |
+
},
|
| 27237 |
+
{
|
| 27238 |
+
"epoch": 3.360858406101622,
|
| 27239 |
+
"eval_loss": 2.2600295543670654,
|
| 27240 |
+
"eval_runtime": 80.7842,
|
| 27241 |
+
"eval_samples_per_second": 61.893,
|
| 27242 |
+
"eval_steps_per_second": 15.473,
|
| 27243 |
+
"num_input_tokens_seen": 87019603744,
|
| 27244 |
+
"step": 153000
|
| 27245 |
}
|
| 27246 |
],
|
| 27247 |
"logging_steps": 50,
|
| 27248 |
"max_steps": 200000,
|
| 27249 |
+
"num_input_tokens_seen": 87019603744,
|
| 27250 |
"num_train_epochs": 5,
|
| 27251 |
"save_steps": 1000,
|
| 27252 |
"stateful_callbacks": {
|
|
|
|
| 27261 |
"attributes": {}
|
| 27262 |
}
|
| 27263 |
},
|
| 27264 |
+
"total_flos": 1.5400882625115718e+20,
|
| 27265 |
"train_batch_size": 32,
|
| 27266 |
"trial_name": null,
|
| 27267 |
"trial_params": null
|