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 150000, 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:374909593e8ea311703df093fe62f4e04001dd17e3c49fbf489dab7fba5c0d5e
|
| 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:8b12ee754bf5fd72453fbeae9c7e6862a347afe5a2e5e26d7a0545500ccc9bf8
|
| 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:75979b1827f6aedbded5556bf5c04453d08a32a155f909439c67d10d01c05c41
|
| 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:305ea49d7e626826ceaad34f0959f2e71b7074c795df15fc38b6daeaea82d52e
|
| 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,
|
|
@@ -26530,11 +26530,189 @@
|
|
| 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":
|
| 26538 |
"num_train_epochs": 5,
|
| 26539 |
"save_steps": 1000,
|
| 26540 |
"stateful_callbacks": {
|
|
@@ -26549,7 +26727,7 @@
|
|
| 26549 |
"attributes": {}
|
| 26550 |
}
|
| 26551 |
},
|
| 26552 |
-
"total_flos": 1.
|
| 26553 |
"train_batch_size": 32,
|
| 26554 |
"trial_name": null,
|
| 26555 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.2949602311378485,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 150000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 26530 |
"eval_steps_per_second": 15.448,
|
| 26531 |
"num_input_tokens_seen": 82825402528,
|
| 26532 |
"step": 149000
|
| 26533 |
+
},
|
| 26534 |
+
{
|
| 26535 |
+
"epoch": 3.2740924757326537,
|
| 26536 |
+
"grad_norm": 0.09636425226926804,
|
| 26537 |
+
"learning_rate": 0.0001,
|
| 26538 |
+
"loss": 2.3512,
|
| 26539 |
+
"num_input_tokens_seen": 82877831328,
|
| 26540 |
+
"step": 149050
|
| 26541 |
+
},
|
| 26542 |
+
{
|
| 26543 |
+
"epoch": 3.275190778648717,
|
| 26544 |
+
"grad_norm": 0.09321967512369156,
|
| 26545 |
+
"learning_rate": 0.0001,
|
| 26546 |
+
"loss": 2.3464,
|
| 26547 |
+
"num_input_tokens_seen": 82930260128,
|
| 26548 |
+
"step": 149100
|
| 26549 |
+
},
|
| 26550 |
+
{
|
| 26551 |
+
"epoch": 3.2762890815647796,
|
| 26552 |
+
"grad_norm": 0.09646619111299515,
|
| 26553 |
+
"learning_rate": 0.0001,
|
| 26554 |
+
"loss": 2.3477,
|
| 26555 |
+
"num_input_tokens_seen": 82982684224,
|
| 26556 |
+
"step": 149150
|
| 26557 |
+
},
|
| 26558 |
+
{
|
| 26559 |
+
"epoch": 3.2773873844808423,
|
| 26560 |
+
"grad_norm": 0.09394371509552002,
|
| 26561 |
+
"learning_rate": 0.0001,
|
| 26562 |
+
"loss": 2.3436,
|
| 26563 |
+
"num_input_tokens_seen": 83035113024,
|
| 26564 |
+
"step": 149200
|
| 26565 |
+
},
|
| 26566 |
+
{
|
| 26567 |
+
"epoch": 3.2784856873969055,
|
| 26568 |
+
"grad_norm": 0.09308401495218277,
|
| 26569 |
+
"learning_rate": 0.0001,
|
| 26570 |
+
"loss": 2.3543,
|
| 26571 |
+
"num_input_tokens_seen": 83087541664,
|
| 26572 |
+
"step": 149250
|
| 26573 |
+
},
|
| 26574 |
+
{
|
| 26575 |
+
"epoch": 3.279583990312968,
|
| 26576 |
+
"grad_norm": 0.09759179502725601,
|
| 26577 |
+
"learning_rate": 0.0001,
|
| 26578 |
+
"loss": 2.3418,
|
| 26579 |
+
"num_input_tokens_seen": 83139969344,
|
| 26580 |
+
"step": 149300
|
| 26581 |
+
},
|
| 26582 |
+
{
|
| 26583 |
+
"epoch": 3.280682293229031,
|
| 26584 |
+
"grad_norm": 0.0901869386434555,
|
| 26585 |
+
"learning_rate": 0.0001,
|
| 26586 |
+
"loss": 2.3484,
|
| 26587 |
+
"num_input_tokens_seen": 83192395872,
|
| 26588 |
+
"step": 149350
|
| 26589 |
+
},
|
| 26590 |
+
{
|
| 26591 |
+
"epoch": 3.281780596145094,
|
| 26592 |
+
"grad_norm": 0.0878322646021843,
|
| 26593 |
+
"learning_rate": 0.0001,
|
| 26594 |
+
"loss": 2.3497,
|
| 26595 |
+
"num_input_tokens_seen": 83244824672,
|
| 26596 |
+
"step": 149400
|
| 26597 |
+
},
|
| 26598 |
+
{
|
| 26599 |
+
"epoch": 3.282878899061157,
|
| 26600 |
+
"grad_norm": 0.09223738312721252,
|
| 26601 |
+
"learning_rate": 0.0001,
|
| 26602 |
+
"loss": 2.3505,
|
| 26603 |
+
"num_input_tokens_seen": 83297245472,
|
| 26604 |
+
"step": 149450
|
| 26605 |
+
},
|
| 26606 |
+
{
|
| 26607 |
+
"epoch": 3.28397720197722,
|
| 26608 |
+
"grad_norm": 0.0960436537861824,
|
| 26609 |
+
"learning_rate": 0.0001,
|
| 26610 |
+
"loss": 2.348,
|
| 26611 |
+
"num_input_tokens_seen": 83349674016,
|
| 26612 |
+
"step": 149500
|
| 26613 |
+
},
|
| 26614 |
+
{
|
| 26615 |
+
"epoch": 3.28397720197722,
|
| 26616 |
+
"eval_loss": 2.2621405124664307,
|
| 26617 |
+
"eval_runtime": 80.8246,
|
| 26618 |
+
"eval_samples_per_second": 61.862,
|
| 26619 |
+
"eval_steps_per_second": 15.466,
|
| 26620 |
+
"num_input_tokens_seen": 83349674016,
|
| 26621 |
+
"step": 149500
|
| 26622 |
+
},
|
| 26623 |
+
{
|
| 26624 |
+
"epoch": 3.2850755048932827,
|
| 26625 |
+
"grad_norm": 0.09093570709228516,
|
| 26626 |
+
"learning_rate": 0.0001,
|
| 26627 |
+
"loss": 2.3531,
|
| 26628 |
+
"num_input_tokens_seen": 83402102304,
|
| 26629 |
+
"step": 149550
|
| 26630 |
+
},
|
| 26631 |
+
{
|
| 26632 |
+
"epoch": 3.2861738078093454,
|
| 26633 |
+
"grad_norm": 0.09345680475234985,
|
| 26634 |
+
"learning_rate": 0.0001,
|
| 26635 |
+
"loss": 2.3425,
|
| 26636 |
+
"num_input_tokens_seen": 83454531104,
|
| 26637 |
+
"step": 149600
|
| 26638 |
+
},
|
| 26639 |
+
{
|
| 26640 |
+
"epoch": 3.2872721107254086,
|
| 26641 |
+
"grad_norm": 0.09687989205121994,
|
| 26642 |
+
"learning_rate": 0.0001,
|
| 26643 |
+
"loss": 2.3467,
|
| 26644 |
+
"num_input_tokens_seen": 83506956512,
|
| 26645 |
+
"step": 149650
|
| 26646 |
+
},
|
| 26647 |
+
{
|
| 26648 |
+
"epoch": 3.2883704136414713,
|
| 26649 |
+
"grad_norm": 0.09994108229875565,
|
| 26650 |
+
"learning_rate": 0.0001,
|
| 26651 |
+
"loss": 2.3526,
|
| 26652 |
+
"num_input_tokens_seen": 83559385312,
|
| 26653 |
+
"step": 149700
|
| 26654 |
+
},
|
| 26655 |
+
{
|
| 26656 |
+
"epoch": 3.2894687165575345,
|
| 26657 |
+
"grad_norm": 0.0894683226943016,
|
| 26658 |
+
"learning_rate": 0.0001,
|
| 26659 |
+
"loss": 2.3545,
|
| 26660 |
+
"num_input_tokens_seen": 83611807424,
|
| 26661 |
+
"step": 149750
|
| 26662 |
+
},
|
| 26663 |
+
{
|
| 26664 |
+
"epoch": 3.290567019473597,
|
| 26665 |
+
"grad_norm": 0.08919432759284973,
|
| 26666 |
+
"learning_rate": 0.0001,
|
| 26667 |
+
"loss": 2.3375,
|
| 26668 |
+
"num_input_tokens_seen": 83664236224,
|
| 26669 |
+
"step": 149800
|
| 26670 |
+
},
|
| 26671 |
+
{
|
| 26672 |
+
"epoch": 3.29166532238966,
|
| 26673 |
+
"grad_norm": 0.0880848690867424,
|
| 26674 |
+
"learning_rate": 0.0001,
|
| 26675 |
+
"loss": 2.3411,
|
| 26676 |
+
"num_input_tokens_seen": 83716662592,
|
| 26677 |
+
"step": 149850
|
| 26678 |
+
},
|
| 26679 |
+
{
|
| 26680 |
+
"epoch": 3.292763625305723,
|
| 26681 |
+
"grad_norm": 0.09474777430295944,
|
| 26682 |
+
"learning_rate": 0.0001,
|
| 26683 |
+
"loss": 2.3454,
|
| 26684 |
+
"num_input_tokens_seen": 83769091072,
|
| 26685 |
+
"step": 149900
|
| 26686 |
+
},
|
| 26687 |
+
{
|
| 26688 |
+
"epoch": 3.293861928221786,
|
| 26689 |
+
"grad_norm": 0.09384060651063919,
|
| 26690 |
+
"learning_rate": 0.0001,
|
| 26691 |
+
"loss": 2.3532,
|
| 26692 |
+
"num_input_tokens_seen": 83821518880,
|
| 26693 |
+
"step": 149950
|
| 26694 |
+
},
|
| 26695 |
+
{
|
| 26696 |
+
"epoch": 3.2949602311378485,
|
| 26697 |
+
"grad_norm": 0.09099478274583817,
|
| 26698 |
+
"learning_rate": 0.0001,
|
| 26699 |
+
"loss": 2.345,
|
| 26700 |
+
"num_input_tokens_seen": 83873945408,
|
| 26701 |
+
"step": 150000
|
| 26702 |
+
},
|
| 26703 |
+
{
|
| 26704 |
+
"epoch": 3.2949602311378485,
|
| 26705 |
+
"eval_loss": 2.2621686458587646,
|
| 26706 |
+
"eval_runtime": 80.2172,
|
| 26707 |
+
"eval_samples_per_second": 62.331,
|
| 26708 |
+
"eval_steps_per_second": 15.583,
|
| 26709 |
+
"num_input_tokens_seen": 83873945408,
|
| 26710 |
+
"step": 150000
|
| 26711 |
}
|
| 26712 |
],
|
| 26713 |
"logging_steps": 50,
|
| 26714 |
"max_steps": 200000,
|
| 26715 |
+
"num_input_tokens_seen": 83873945408,
|
| 26716 |
"num_train_epochs": 5,
|
| 26717 |
"save_steps": 1000,
|
| 26718 |
"stateful_callbacks": {
|
|
|
|
| 26727 |
"attributes": {}
|
| 26728 |
}
|
| 26729 |
},
|
| 26730 |
+
"total_flos": 1.484415847645176e+20,
|
| 26731 |
"train_batch_size": 32,
|
| 26732 |
"trial_name": null,
|
| 26733 |
"trial_params": null
|