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 155000, 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:e85428a0f815969dced7abfb0c3db479ab6e9ac0b0476ec45794af0f2488296e
|
| 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:4cfe591240bbf8339bc0b20dc4896e7920caa6915b0ec035cadf1d16c6737411
|
| 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:13333a98c36f942bf81dc162e3112c4a9758770e0e5d231e4e8af488e2291291
|
| 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:46ed0a6739f44065f2f96ef3493947a9f7597f7e2d294443183887f798d35555
|
| 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,
|
|
@@ -27420,11 +27420,189 @@
|
|
| 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":
|
| 27428 |
"num_train_epochs": 5,
|
| 27429 |
"save_steps": 1000,
|
| 27430 |
"stateful_callbacks": {
|
|
@@ -27439,7 +27617,7 @@
|
|
| 27439 |
"attributes": {}
|
| 27440 |
}
|
| 27441 |
},
|
| 27442 |
-
"total_flos": 1.
|
| 27443 |
"train_batch_size": 32,
|
| 27444 |
"trial_name": null,
|
| 27445 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 3.404790522744137,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 155000,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 27420 |
"eval_steps_per_second": 15.483,
|
| 27421 |
"num_input_tokens_seen": 88068164512,
|
| 27422 |
"step": 154000
|
| 27423 |
+
},
|
| 27424 |
+
{
|
| 27425 |
+
"epoch": 3.3839227673389427,
|
| 27426 |
+
"grad_norm": 0.09020719677209854,
|
| 27427 |
+
"learning_rate": 0.0001,
|
| 27428 |
+
"loss": 2.3497,
|
| 27429 |
+
"num_input_tokens_seen": 88120592672,
|
| 27430 |
+
"step": 154050
|
| 27431 |
+
},
|
| 27432 |
+
{
|
| 27433 |
+
"epoch": 3.3850210702550054,
|
| 27434 |
+
"grad_norm": 0.09590700268745422,
|
| 27435 |
+
"learning_rate": 0.0001,
|
| 27436 |
+
"loss": 2.3511,
|
| 27437 |
+
"num_input_tokens_seen": 88173020768,
|
| 27438 |
+
"step": 154100
|
| 27439 |
+
},
|
| 27440 |
+
{
|
| 27441 |
+
"epoch": 3.386119373171068,
|
| 27442 |
+
"grad_norm": 0.09949789196252823,
|
| 27443 |
+
"learning_rate": 0.0001,
|
| 27444 |
+
"loss": 2.3442,
|
| 27445 |
+
"num_input_tokens_seen": 88225449568,
|
| 27446 |
+
"step": 154150
|
| 27447 |
+
},
|
| 27448 |
+
{
|
| 27449 |
+
"epoch": 3.387217676087131,
|
| 27450 |
+
"grad_norm": 0.08850202709436417,
|
| 27451 |
+
"learning_rate": 0.0001,
|
| 27452 |
+
"loss": 2.3491,
|
| 27453 |
+
"num_input_tokens_seen": 88277876544,
|
| 27454 |
+
"step": 154200
|
| 27455 |
+
},
|
| 27456 |
+
{
|
| 27457 |
+
"epoch": 3.388315979003194,
|
| 27458 |
+
"grad_norm": 0.09314405918121338,
|
| 27459 |
+
"learning_rate": 0.0001,
|
| 27460 |
+
"loss": 2.3502,
|
| 27461 |
+
"num_input_tokens_seen": 88330304416,
|
| 27462 |
+
"step": 154250
|
| 27463 |
+
},
|
| 27464 |
+
{
|
| 27465 |
+
"epoch": 3.3894142819192568,
|
| 27466 |
+
"grad_norm": 0.09926445782184601,
|
| 27467 |
+
"learning_rate": 0.0001,
|
| 27468 |
+
"loss": 2.345,
|
| 27469 |
+
"num_input_tokens_seen": 88382733216,
|
| 27470 |
+
"step": 154300
|
| 27471 |
+
},
|
| 27472 |
+
{
|
| 27473 |
+
"epoch": 3.39051258483532,
|
| 27474 |
+
"grad_norm": 0.09468773007392883,
|
| 27475 |
+
"learning_rate": 0.0001,
|
| 27476 |
+
"loss": 2.3429,
|
| 27477 |
+
"num_input_tokens_seen": 88435156832,
|
| 27478 |
+
"step": 154350
|
| 27479 |
+
},
|
| 27480 |
+
{
|
| 27481 |
+
"epoch": 3.3916108877513826,
|
| 27482 |
+
"grad_norm": 0.09661844372749329,
|
| 27483 |
+
"learning_rate": 0.0001,
|
| 27484 |
+
"loss": 2.3465,
|
| 27485 |
+
"num_input_tokens_seen": 88487585632,
|
| 27486 |
+
"step": 154400
|
| 27487 |
+
},
|
| 27488 |
+
{
|
| 27489 |
+
"epoch": 3.3927091906674454,
|
| 27490 |
+
"grad_norm": 0.09253469109535217,
|
| 27491 |
+
"learning_rate": 0.0001,
|
| 27492 |
+
"loss": 2.3455,
|
| 27493 |
+
"num_input_tokens_seen": 88540012544,
|
| 27494 |
+
"step": 154450
|
| 27495 |
+
},
|
| 27496 |
+
{
|
| 27497 |
+
"epoch": 3.3938074935835085,
|
| 27498 |
+
"grad_norm": 0.08921549469232559,
|
| 27499 |
+
"learning_rate": 0.0001,
|
| 27500 |
+
"loss": 2.3491,
|
| 27501 |
+
"num_input_tokens_seen": 88592441344,
|
| 27502 |
+
"step": 154500
|
| 27503 |
+
},
|
| 27504 |
+
{
|
| 27505 |
+
"epoch": 3.3938074935835085,
|
| 27506 |
+
"eval_loss": 2.259216785430908,
|
| 27507 |
+
"eval_runtime": 80.6571,
|
| 27508 |
+
"eval_samples_per_second": 61.991,
|
| 27509 |
+
"eval_steps_per_second": 15.498,
|
| 27510 |
+
"num_input_tokens_seen": 88592441344,
|
| 27511 |
+
"step": 154500
|
| 27512 |
+
},
|
| 27513 |
+
{
|
| 27514 |
+
"epoch": 3.3949057964995712,
|
| 27515 |
+
"grad_norm": 0.09509548544883728,
|
| 27516 |
+
"learning_rate": 0.0001,
|
| 27517 |
+
"loss": 2.3423,
|
| 27518 |
+
"num_input_tokens_seen": 88644870144,
|
| 27519 |
+
"step": 154550
|
| 27520 |
+
},
|
| 27521 |
+
{
|
| 27522 |
+
"epoch": 3.3960040994156344,
|
| 27523 |
+
"grad_norm": 0.0929824560880661,
|
| 27524 |
+
"learning_rate": 0.0001,
|
| 27525 |
+
"loss": 2.3429,
|
| 27526 |
+
"num_input_tokens_seen": 88697298944,
|
| 27527 |
+
"step": 154600
|
| 27528 |
+
},
|
| 27529 |
+
{
|
| 27530 |
+
"epoch": 3.397102402331697,
|
| 27531 |
+
"grad_norm": 0.09483112394809723,
|
| 27532 |
+
"learning_rate": 0.0001,
|
| 27533 |
+
"loss": 2.3451,
|
| 27534 |
+
"num_input_tokens_seen": 88749721920,
|
| 27535 |
+
"step": 154650
|
| 27536 |
+
},
|
| 27537 |
+
{
|
| 27538 |
+
"epoch": 3.39820070524776,
|
| 27539 |
+
"grad_norm": 0.09392872452735901,
|
| 27540 |
+
"learning_rate": 0.0001,
|
| 27541 |
+
"loss": 2.3437,
|
| 27542 |
+
"num_input_tokens_seen": 88802150720,
|
| 27543 |
+
"step": 154700
|
| 27544 |
+
},
|
| 27545 |
+
{
|
| 27546 |
+
"epoch": 3.399299008163823,
|
| 27547 |
+
"grad_norm": 0.09336952865123749,
|
| 27548 |
+
"learning_rate": 0.0001,
|
| 27549 |
+
"loss": 2.3414,
|
| 27550 |
+
"num_input_tokens_seen": 88854579520,
|
| 27551 |
+
"step": 154750
|
| 27552 |
+
},
|
| 27553 |
+
{
|
| 27554 |
+
"epoch": 3.4003973110798857,
|
| 27555 |
+
"grad_norm": 0.0942087471485138,
|
| 27556 |
+
"learning_rate": 0.0001,
|
| 27557 |
+
"loss": 2.3497,
|
| 27558 |
+
"num_input_tokens_seen": 88907008320,
|
| 27559 |
+
"step": 154800
|
| 27560 |
+
},
|
| 27561 |
+
{
|
| 27562 |
+
"epoch": 3.4014956139959485,
|
| 27563 |
+
"grad_norm": 0.10055429488420486,
|
| 27564 |
+
"learning_rate": 0.0001,
|
| 27565 |
+
"loss": 2.3472,
|
| 27566 |
+
"num_input_tokens_seen": 88959437120,
|
| 27567 |
+
"step": 154850
|
| 27568 |
+
},
|
| 27569 |
+
{
|
| 27570 |
+
"epoch": 3.4025939169120116,
|
| 27571 |
+
"grad_norm": 0.10026364773511887,
|
| 27572 |
+
"learning_rate": 0.0001,
|
| 27573 |
+
"loss": 2.3524,
|
| 27574 |
+
"num_input_tokens_seen": 89011865920,
|
| 27575 |
+
"step": 154900
|
| 27576 |
+
},
|
| 27577 |
+
{
|
| 27578 |
+
"epoch": 3.4036922198280744,
|
| 27579 |
+
"grad_norm": 0.09667252004146576,
|
| 27580 |
+
"learning_rate": 0.0001,
|
| 27581 |
+
"loss": 2.3448,
|
| 27582 |
+
"num_input_tokens_seen": 89064293312,
|
| 27583 |
+
"step": 154950
|
| 27584 |
+
},
|
| 27585 |
+
{
|
| 27586 |
+
"epoch": 3.404790522744137,
|
| 27587 |
+
"grad_norm": 3.4680259227752686,
|
| 27588 |
+
"learning_rate": 0.0001,
|
| 27589 |
+
"loss": 2.3532,
|
| 27590 |
+
"num_input_tokens_seen": 89116722112,
|
| 27591 |
+
"step": 155000
|
| 27592 |
+
},
|
| 27593 |
+
{
|
| 27594 |
+
"epoch": 3.404790522744137,
|
| 27595 |
+
"eval_loss": 2.2594308853149414,
|
| 27596 |
+
"eval_runtime": 80.9279,
|
| 27597 |
+
"eval_samples_per_second": 61.783,
|
| 27598 |
+
"eval_steps_per_second": 15.446,
|
| 27599 |
+
"num_input_tokens_seen": 89116722112,
|
| 27600 |
+
"step": 155000
|
| 27601 |
}
|
| 27602 |
],
|
| 27603 |
"logging_steps": 50,
|
| 27604 |
"max_steps": 200000,
|
| 27605 |
+
"num_input_tokens_seen": 89116722112,
|
| 27606 |
"num_train_epochs": 5,
|
| 27607 |
"save_steps": 1000,
|
| 27608 |
"stateful_callbacks": {
|
|
|
|
| 27617 |
"attributes": {}
|
| 27618 |
}
|
| 27619 |
},
|
| 27620 |
+
"total_flos": 1.5772034324812677e+20,
|
| 27621 |
"train_batch_size": 32,
|
| 27622 |
"trial_name": null,
|
| 27623 |
"trial_params": null
|