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 149000, 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:0b87dded127d40fd82c03235b1698dfe33a8b5f87187ba074bf40022236624c6
|
| 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:23c47d38cecaa63e531df29e38e9afb0727c11e9a4b57b059ee2acbb0fb1f4d3
|
| 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:4b3f339c8acfe7277fe2df77c27fc1a5816faed78935e9068ab888b7bb106bcc
|
| 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:c40e8b457d57718c67b26e249dafca477bb77feb974e04c20b75eee25d0b14aa
|
| 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,
|
|
@@ -26352,11 +26352,189 @@
|
|
| 26352 |
"eval_steps_per_second": 15.617,
|
| 26353 |
"num_input_tokens_seen": 81776845856,
|
| 26354 |
"step": 148000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26355 |
}
|
| 26356 |
],
|
| 26357 |
"logging_steps": 50,
|
| 26358 |
"max_steps": 200000,
|
| 26359 |
-
"num_input_tokens_seen":
|
| 26360 |
"num_train_epochs": 5,
|
| 26361 |
"save_steps": 1000,
|
| 26362 |
"stateful_callbacks": {
|
|
@@ -26371,7 +26549,7 @@
|
|
| 26371 |
"attributes": {}
|
| 26372 |
}
|
| 26373 |
},
|
| 26374 |
-
"total_flos": 1.
|
| 26375 |
"train_batch_size": 32,
|
| 26376 |
"trial_name": null,
|
| 26377 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.272994172816591,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 149000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 26352 |
"eval_steps_per_second": 15.617,
|
| 26353 |
"num_input_tokens_seen": 81776845856,
|
| 26354 |
"step": 148000
|
| 26355 |
+
},
|
| 26356 |
+
{
|
| 26357 |
+
"epoch": 3.252126417411396,
|
| 26358 |
+
"grad_norm": 0.09104160964488983,
|
| 26359 |
+
"learning_rate": 0.0001,
|
| 26360 |
+
"loss": 2.3493,
|
| 26361 |
+
"num_input_tokens_seen": 81829274656,
|
| 26362 |
+
"step": 148050
|
| 26363 |
+
},
|
| 26364 |
+
{
|
| 26365 |
+
"epoch": 3.253224720327459,
|
| 26366 |
+
"grad_norm": 0.09389001131057739,
|
| 26367 |
+
"learning_rate": 0.0001,
|
| 26368 |
+
"loss": 2.3517,
|
| 26369 |
+
"num_input_tokens_seen": 81881698752,
|
| 26370 |
+
"step": 148100
|
| 26371 |
+
},
|
| 26372 |
+
{
|
| 26373 |
+
"epoch": 3.254323023243522,
|
| 26374 |
+
"grad_norm": 0.09085691720247269,
|
| 26375 |
+
"learning_rate": 0.0001,
|
| 26376 |
+
"loss": 2.3449,
|
| 26377 |
+
"num_input_tokens_seen": 81934124352,
|
| 26378 |
+
"step": 148150
|
| 26379 |
+
},
|
| 26380 |
+
{
|
| 26381 |
+
"epoch": 3.255421326159585,
|
| 26382 |
+
"grad_norm": 0.09462492913007736,
|
| 26383 |
+
"learning_rate": 0.0001,
|
| 26384 |
+
"loss": 2.3535,
|
| 26385 |
+
"num_input_tokens_seen": 81986553152,
|
| 26386 |
+
"step": 148200
|
| 26387 |
+
},
|
| 26388 |
+
{
|
| 26389 |
+
"epoch": 3.2565196290756475,
|
| 26390 |
+
"grad_norm": 0.09079829603433609,
|
| 26391 |
+
"learning_rate": 0.0001,
|
| 26392 |
+
"loss": 2.3477,
|
| 26393 |
+
"num_input_tokens_seen": 82038981952,
|
| 26394 |
+
"step": 148250
|
| 26395 |
+
},
|
| 26396 |
+
{
|
| 26397 |
+
"epoch": 3.2576179319917107,
|
| 26398 |
+
"grad_norm": 0.09165850281715393,
|
| 26399 |
+
"learning_rate": 0.0001,
|
| 26400 |
+
"loss": 2.3548,
|
| 26401 |
+
"num_input_tokens_seen": 82091407040,
|
| 26402 |
+
"step": 148300
|
| 26403 |
+
},
|
| 26404 |
+
{
|
| 26405 |
+
"epoch": 3.2587162349077734,
|
| 26406 |
+
"grad_norm": 0.09388460963964462,
|
| 26407 |
+
"learning_rate": 0.0001,
|
| 26408 |
+
"loss": 2.3477,
|
| 26409 |
+
"num_input_tokens_seen": 82143834912,
|
| 26410 |
+
"step": 148350
|
| 26411 |
+
},
|
| 26412 |
+
{
|
| 26413 |
+
"epoch": 3.2598145378238366,
|
| 26414 |
+
"grad_norm": 0.09309230744838715,
|
| 26415 |
+
"learning_rate": 0.0001,
|
| 26416 |
+
"loss": 2.3506,
|
| 26417 |
+
"num_input_tokens_seen": 82196262432,
|
| 26418 |
+
"step": 148400
|
| 26419 |
+
},
|
| 26420 |
+
{
|
| 26421 |
+
"epoch": 3.2609128407398993,
|
| 26422 |
+
"grad_norm": 0.09229093790054321,
|
| 26423 |
+
"learning_rate": 0.0001,
|
| 26424 |
+
"loss": 2.3451,
|
| 26425 |
+
"num_input_tokens_seen": 82248691232,
|
| 26426 |
+
"step": 148450
|
| 26427 |
+
},
|
| 26428 |
+
{
|
| 26429 |
+
"epoch": 3.262011143655962,
|
| 26430 |
+
"grad_norm": 0.09784185886383057,
|
| 26431 |
+
"learning_rate": 0.0001,
|
| 26432 |
+
"loss": 2.3447,
|
| 26433 |
+
"num_input_tokens_seen": 82301120032,
|
| 26434 |
+
"step": 148500
|
| 26435 |
+
},
|
| 26436 |
+
{
|
| 26437 |
+
"epoch": 3.262011143655962,
|
| 26438 |
+
"eval_loss": 2.2628486156463623,
|
| 26439 |
+
"eval_runtime": 80.5596,
|
| 26440 |
+
"eval_samples_per_second": 62.066,
|
| 26441 |
+
"eval_steps_per_second": 15.516,
|
| 26442 |
+
"num_input_tokens_seen": 82301120032,
|
| 26443 |
+
"step": 148500
|
| 26444 |
+
},
|
| 26445 |
+
{
|
| 26446 |
+
"epoch": 3.263109446572025,
|
| 26447 |
+
"grad_norm": 0.09117227792739868,
|
| 26448 |
+
"learning_rate": 0.0001,
|
| 26449 |
+
"loss": 2.3541,
|
| 26450 |
+
"num_input_tokens_seen": 82353545696,
|
| 26451 |
+
"step": 148550
|
| 26452 |
+
},
|
| 26453 |
+
{
|
| 26454 |
+
"epoch": 3.264207749488088,
|
| 26455 |
+
"grad_norm": 0.09306716918945312,
|
| 26456 |
+
"learning_rate": 0.0001,
|
| 26457 |
+
"loss": 2.3522,
|
| 26458 |
+
"num_input_tokens_seen": 82405974496,
|
| 26459 |
+
"step": 148600
|
| 26460 |
+
},
|
| 26461 |
+
{
|
| 26462 |
+
"epoch": 3.2653060524041506,
|
| 26463 |
+
"grad_norm": 0.0911402553319931,
|
| 26464 |
+
"learning_rate": 0.0001,
|
| 26465 |
+
"loss": 2.3479,
|
| 26466 |
+
"num_input_tokens_seen": 82458403296,
|
| 26467 |
+
"step": 148650
|
| 26468 |
+
},
|
| 26469 |
+
{
|
| 26470 |
+
"epoch": 3.266404355320214,
|
| 26471 |
+
"grad_norm": 0.09247133135795593,
|
| 26472 |
+
"learning_rate": 0.0001,
|
| 26473 |
+
"loss": 2.3459,
|
| 26474 |
+
"num_input_tokens_seen": 82510832096,
|
| 26475 |
+
"step": 148700
|
| 26476 |
+
},
|
| 26477 |
+
{
|
| 26478 |
+
"epoch": 3.2675026582362765,
|
| 26479 |
+
"grad_norm": 0.08942971378564835,
|
| 26480 |
+
"learning_rate": 0.0001,
|
| 26481 |
+
"loss": 2.3442,
|
| 26482 |
+
"num_input_tokens_seen": 82563260896,
|
| 26483 |
+
"step": 148750
|
| 26484 |
+
},
|
| 26485 |
+
{
|
| 26486 |
+
"epoch": 3.2686009611523392,
|
| 26487 |
+
"grad_norm": 0.09245148301124573,
|
| 26488 |
+
"learning_rate": 0.0001,
|
| 26489 |
+
"loss": 2.3455,
|
| 26490 |
+
"num_input_tokens_seen": 82615689600,
|
| 26491 |
+
"step": 148800
|
| 26492 |
+
},
|
| 26493 |
+
{
|
| 26494 |
+
"epoch": 3.2696992640684024,
|
| 26495 |
+
"grad_norm": 0.09015721827745438,
|
| 26496 |
+
"learning_rate": 0.0001,
|
| 26497 |
+
"loss": 2.3502,
|
| 26498 |
+
"num_input_tokens_seen": 82668116128,
|
| 26499 |
+
"step": 148850
|
| 26500 |
+
},
|
| 26501 |
+
{
|
| 26502 |
+
"epoch": 3.270797566984465,
|
| 26503 |
+
"grad_norm": 0.1070784255862236,
|
| 26504 |
+
"learning_rate": 0.0001,
|
| 26505 |
+
"loss": 2.3561,
|
| 26506 |
+
"num_input_tokens_seen": 82720544928,
|
| 26507 |
+
"step": 148900
|
| 26508 |
+
},
|
| 26509 |
+
{
|
| 26510 |
+
"epoch": 3.2718958699005283,
|
| 26511 |
+
"grad_norm": 0.09262741357088089,
|
| 26512 |
+
"learning_rate": 0.0001,
|
| 26513 |
+
"loss": 2.343,
|
| 26514 |
+
"num_input_tokens_seen": 82772973728,
|
| 26515 |
+
"step": 148950
|
| 26516 |
+
},
|
| 26517 |
+
{
|
| 26518 |
+
"epoch": 3.272994172816591,
|
| 26519 |
+
"grad_norm": 0.0885642021894455,
|
| 26520 |
+
"learning_rate": 0.0001,
|
| 26521 |
+
"loss": 2.3477,
|
| 26522 |
+
"num_input_tokens_seen": 82825402528,
|
| 26523 |
+
"step": 149000
|
| 26524 |
+
},
|
| 26525 |
+
{
|
| 26526 |
+
"epoch": 3.272994172816591,
|
| 26527 |
+
"eval_loss": 2.262723684310913,
|
| 26528 |
+
"eval_runtime": 80.9161,
|
| 26529 |
+
"eval_samples_per_second": 61.792,
|
| 26530 |
+
"eval_steps_per_second": 15.448,
|
| 26531 |
+
"num_input_tokens_seen": 82825402528,
|
| 26532 |
+
"step": 149000
|
| 26533 |
}
|
| 26534 |
],
|
| 26535 |
"logging_steps": 50,
|
| 26536 |
"max_steps": 200000,
|
| 26537 |
+
"num_input_tokens_seen": 82825402528,
|
| 26538 |
"num_train_epochs": 5,
|
| 26539 |
"save_steps": 1000,
|
| 26540 |
"stateful_callbacks": {
|
|
|
|
| 26549 |
"attributes": {}
|
| 26550 |
}
|
| 26551 |
},
|
| 26552 |
+
"total_flos": 1.465858551211389e+20,
|
| 26553 |
"train_batch_size": 32,
|
| 26554 |
"trial_name": null,
|
| 26555 |
"trial_params": null
|