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 154000, 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:e6bcc61c9f454b7a64e29716235e1908b8ccfcc57499ab5454255b1bfbcadea7
|
| 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:4002d60f1e4a15c1f11ee63cc36aa183850a72f44f466a3caa743e3506211571
|
| 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:4084d1a8f5942c88851794ec5108b468d411ca529332c3f218a8ddcc004c0a0e
|
| 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:74dd121f56ba98a31c6c3f650af9c804c1ddf5d4b98b55654ba270708d2e7766
|
| 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,
|
|
@@ -27242,11 +27242,189 @@
|
|
| 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":
|
| 27250 |
"num_train_epochs": 5,
|
| 27251 |
"save_steps": 1000,
|
| 27252 |
"stateful_callbacks": {
|
|
@@ -27261,7 +27439,7 @@
|
|
| 27261 |
"attributes": {}
|
| 27262 |
}
|
| 27263 |
},
|
| 27264 |
-
"total_flos": 1.
|
| 27265 |
"train_batch_size": 32,
|
| 27266 |
"trial_name": null,
|
| 27267 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.3828244644228795,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 154000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 27242 |
"eval_steps_per_second": 15.473,
|
| 27243 |
"num_input_tokens_seen": 87019603744,
|
| 27244 |
"step": 153000
|
| 27245 |
+
},
|
| 27246 |
+
{
|
| 27247 |
+
"epoch": 3.3619567090176847,
|
| 27248 |
+
"grad_norm": 0.09880559146404266,
|
| 27249 |
+
"learning_rate": 0.0001,
|
| 27250 |
+
"loss": 2.3507,
|
| 27251 |
+
"num_input_tokens_seen": 87072028928,
|
| 27252 |
+
"step": 153050
|
| 27253 |
+
},
|
| 27254 |
+
{
|
| 27255 |
+
"epoch": 3.3630550119337475,
|
| 27256 |
+
"grad_norm": 0.09557171911001205,
|
| 27257 |
+
"learning_rate": 0.0001,
|
| 27258 |
+
"loss": 2.3501,
|
| 27259 |
+
"num_input_tokens_seen": 87124457728,
|
| 27260 |
+
"step": 153100
|
| 27261 |
+
},
|
| 27262 |
+
{
|
| 27263 |
+
"epoch": 3.3641533148498106,
|
| 27264 |
+
"grad_norm": 0.09662795066833496,
|
| 27265 |
+
"learning_rate": 0.0001,
|
| 27266 |
+
"loss": 2.3499,
|
| 27267 |
+
"num_input_tokens_seen": 87176886528,
|
| 27268 |
+
"step": 153150
|
| 27269 |
+
},
|
| 27270 |
+
{
|
| 27271 |
+
"epoch": 3.3652516177658733,
|
| 27272 |
+
"grad_norm": 0.09444860368967056,
|
| 27273 |
+
"learning_rate": 0.0001,
|
| 27274 |
+
"loss": 2.3455,
|
| 27275 |
+
"num_input_tokens_seen": 87229314144,
|
| 27276 |
+
"step": 153200
|
| 27277 |
+
},
|
| 27278 |
+
{
|
| 27279 |
+
"epoch": 3.3663499206819365,
|
| 27280 |
+
"grad_norm": 0.09235711395740509,
|
| 27281 |
+
"learning_rate": 0.0001,
|
| 27282 |
+
"loss": 2.3414,
|
| 27283 |
+
"num_input_tokens_seen": 87281742944,
|
| 27284 |
+
"step": 153250
|
| 27285 |
+
},
|
| 27286 |
+
{
|
| 27287 |
+
"epoch": 3.3674482235979992,
|
| 27288 |
+
"grad_norm": 0.09402730315923691,
|
| 27289 |
+
"learning_rate": 0.0001,
|
| 27290 |
+
"loss": 2.3476,
|
| 27291 |
+
"num_input_tokens_seen": 87334170144,
|
| 27292 |
+
"step": 153300
|
| 27293 |
+
},
|
| 27294 |
+
{
|
| 27295 |
+
"epoch": 3.368546526514062,
|
| 27296 |
+
"grad_norm": 0.09448970854282379,
|
| 27297 |
+
"learning_rate": 0.0001,
|
| 27298 |
+
"loss": 2.3455,
|
| 27299 |
+
"num_input_tokens_seen": 87386598944,
|
| 27300 |
+
"step": 153350
|
| 27301 |
+
},
|
| 27302 |
+
{
|
| 27303 |
+
"epoch": 3.369644829430125,
|
| 27304 |
+
"grad_norm": 0.09693534672260284,
|
| 27305 |
+
"learning_rate": 0.0001,
|
| 27306 |
+
"loss": 2.3438,
|
| 27307 |
+
"num_input_tokens_seen": 87439027744,
|
| 27308 |
+
"step": 153400
|
| 27309 |
+
},
|
| 27310 |
+
{
|
| 27311 |
+
"epoch": 3.370743132346188,
|
| 27312 |
+
"grad_norm": 0.09623382240533829,
|
| 27313 |
+
"learning_rate": 0.0001,
|
| 27314 |
+
"loss": 2.3441,
|
| 27315 |
+
"num_input_tokens_seen": 87491455136,
|
| 27316 |
+
"step": 153450
|
| 27317 |
+
},
|
| 27318 |
+
{
|
| 27319 |
+
"epoch": 3.3718414352622506,
|
| 27320 |
+
"grad_norm": 0.09151625633239746,
|
| 27321 |
+
"learning_rate": 0.0001,
|
| 27322 |
+
"loss": 2.3437,
|
| 27323 |
+
"num_input_tokens_seen": 87543883936,
|
| 27324 |
+
"step": 153500
|
| 27325 |
+
},
|
| 27326 |
+
{
|
| 27327 |
+
"epoch": 3.3718414352622506,
|
| 27328 |
+
"eval_loss": 2.2598507404327393,
|
| 27329 |
+
"eval_runtime": 80.5747,
|
| 27330 |
+
"eval_samples_per_second": 62.054,
|
| 27331 |
+
"eval_steps_per_second": 15.514,
|
| 27332 |
+
"num_input_tokens_seen": 87543883936,
|
| 27333 |
+
"step": 153500
|
| 27334 |
+
},
|
| 27335 |
+
{
|
| 27336 |
+
"epoch": 3.3729397381783137,
|
| 27337 |
+
"grad_norm": 0.09348684549331665,
|
| 27338 |
+
"learning_rate": 0.0001,
|
| 27339 |
+
"loss": 2.352,
|
| 27340 |
+
"num_input_tokens_seen": 87596312736,
|
| 27341 |
+
"step": 153550
|
| 27342 |
+
},
|
| 27343 |
+
{
|
| 27344 |
+
"epoch": 3.3740380410943764,
|
| 27345 |
+
"grad_norm": 0.08888132125139236,
|
| 27346 |
+
"learning_rate": 0.0001,
|
| 27347 |
+
"loss": 2.3439,
|
| 27348 |
+
"num_input_tokens_seen": 87648741536,
|
| 27349 |
+
"step": 153600
|
| 27350 |
+
},
|
| 27351 |
+
{
|
| 27352 |
+
"epoch": 3.375136344010439,
|
| 27353 |
+
"grad_norm": 0.09428702294826508,
|
| 27354 |
+
"learning_rate": 0.0001,
|
| 27355 |
+
"loss": 2.3515,
|
| 27356 |
+
"num_input_tokens_seen": 87701167552,
|
| 27357 |
+
"step": 153650
|
| 27358 |
+
},
|
| 27359 |
+
{
|
| 27360 |
+
"epoch": 3.3762346469265023,
|
| 27361 |
+
"grad_norm": 0.0958220586180687,
|
| 27362 |
+
"learning_rate": 0.0001,
|
| 27363 |
+
"loss": 2.3509,
|
| 27364 |
+
"num_input_tokens_seen": 87753593472,
|
| 27365 |
+
"step": 153700
|
| 27366 |
+
},
|
| 27367 |
+
{
|
| 27368 |
+
"epoch": 3.377332949842565,
|
| 27369 |
+
"grad_norm": 0.0915323942899704,
|
| 27370 |
+
"learning_rate": 0.0001,
|
| 27371 |
+
"loss": 2.3424,
|
| 27372 |
+
"num_input_tokens_seen": 87806022112,
|
| 27373 |
+
"step": 153750
|
| 27374 |
+
},
|
| 27375 |
+
{
|
| 27376 |
+
"epoch": 3.378431252758628,
|
| 27377 |
+
"grad_norm": 0.09283678978681564,
|
| 27378 |
+
"learning_rate": 0.0001,
|
| 27379 |
+
"loss": 2.3523,
|
| 27380 |
+
"num_input_tokens_seen": 87858450912,
|
| 27381 |
+
"step": 153800
|
| 27382 |
+
},
|
| 27383 |
+
{
|
| 27384 |
+
"epoch": 3.379529555674691,
|
| 27385 |
+
"grad_norm": 0.09269370883703232,
|
| 27386 |
+
"learning_rate": 0.0001,
|
| 27387 |
+
"loss": 2.348,
|
| 27388 |
+
"num_input_tokens_seen": 87910879712,
|
| 27389 |
+
"step": 153850
|
| 27390 |
+
},
|
| 27391 |
+
{
|
| 27392 |
+
"epoch": 3.3806278585907537,
|
| 27393 |
+
"grad_norm": 0.09313447028398514,
|
| 27394 |
+
"learning_rate": 0.0001,
|
| 27395 |
+
"loss": 2.3479,
|
| 27396 |
+
"num_input_tokens_seen": 87963308512,
|
| 27397 |
+
"step": 153900
|
| 27398 |
+
},
|
| 27399 |
+
{
|
| 27400 |
+
"epoch": 3.381726161506817,
|
| 27401 |
+
"grad_norm": 0.09186484664678574,
|
| 27402 |
+
"learning_rate": 0.0001,
|
| 27403 |
+
"loss": 2.3438,
|
| 27404 |
+
"num_input_tokens_seen": 88015735712,
|
| 27405 |
+
"step": 153950
|
| 27406 |
+
},
|
| 27407 |
+
{
|
| 27408 |
+
"epoch": 3.3828244644228795,
|
| 27409 |
+
"grad_norm": 0.09128747135400772,
|
| 27410 |
+
"learning_rate": 0.0001,
|
| 27411 |
+
"loss": 2.3461,
|
| 27412 |
+
"num_input_tokens_seen": 88068164512,
|
| 27413 |
+
"step": 154000
|
| 27414 |
+
},
|
| 27415 |
+
{
|
| 27416 |
+
"epoch": 3.3828244644228795,
|
| 27417 |
+
"eval_loss": 2.259556293487549,
|
| 27418 |
+
"eval_runtime": 80.7318,
|
| 27419 |
+
"eval_samples_per_second": 61.933,
|
| 27420 |
+
"eval_steps_per_second": 15.483,
|
| 27421 |
+
"num_input_tokens_seen": 88068164512,
|
| 27422 |
+
"step": 154000
|
| 27423 |
}
|
| 27424 |
],
|
| 27425 |
"logging_steps": 50,
|
| 27426 |
"max_steps": 200000,
|
| 27427 |
+
"num_input_tokens_seen": 88068164512,
|
| 27428 |
"num_train_epochs": 5,
|
| 27429 |
"save_steps": 1000,
|
| 27430 |
"stateful_callbacks": {
|
|
|
|
| 27439 |
"attributes": {}
|
| 27440 |
}
|
| 27441 |
},
|
| 27442 |
+
"total_flos": 1.5586458755303306e+20,
|
| 27443 |
"train_batch_size": 32,
|
| 27444 |
"trial_name": null,
|
| 27445 |
"trial_params": null
|