Text Generation
Transformers
Safetensors
English
qwen2
atlas-coder
qwen2.5-coder
sub-1b
qlora
code-generation
code-completion
code-debugging
fine-tuned
text-generation-inference
conversational
Instructions to use Pluto-AI-Labs/Atlas-Coder-0.5B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Pluto-AI-Labs/Atlas-Coder-0.5B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Pluto-AI-Labs/Atlas-Coder-0.5B") messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("Pluto-AI-Labs/Atlas-Coder-0.5B") model = AutoModelForCausalLM.from_pretrained("Pluto-AI-Labs/Atlas-Coder-0.5B", device_map="auto") messages = [ {"role": "user", "content": "Who are you?"}, ] inputs = tokenizer.apply_chat_template( messages, add_generation_prompt=True, tokenize=True, return_dict=True, return_tensors="pt", ).to(model.device) outputs = model.generate(**inputs, max_new_tokens=40) print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:])) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use Pluto-AI-Labs/Atlas-Coder-0.5B with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "Pluto-AI-Labs/Atlas-Coder-0.5B" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Pluto-AI-Labs/Atlas-Coder-0.5B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/Pluto-AI-Labs/Atlas-Coder-0.5B
- SGLang
How to use Pluto-AI-Labs/Atlas-Coder-0.5B 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 "Pluto-AI-Labs/Atlas-Coder-0.5B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Pluto-AI-Labs/Atlas-Coder-0.5B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'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 "Pluto-AI-Labs/Atlas-Coder-0.5B" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Pluto-AI-Labs/Atlas-Coder-0.5B", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use Pluto-AI-Labs/Atlas-Coder-0.5B with Docker Model Runner:
docker model run hf.co/Pluto-AI-Labs/Atlas-Coder-0.5B
Training in progress, step 3500, checkpoint
Browse files
last-checkpoint/adapter_config.json
CHANGED
|
@@ -25,13 +25,13 @@
|
|
| 25 |
"rank_pattern": {},
|
| 26 |
"revision": null,
|
| 27 |
"target_modules": [
|
| 28 |
-
"down_proj",
|
| 29 |
-
"up_proj",
|
| 30 |
"v_proj",
|
|
|
|
|
|
|
| 31 |
"q_proj",
|
| 32 |
-
"o_proj",
|
| 33 |
"k_proj",
|
| 34 |
-
"gate_proj"
|
|
|
|
| 35 |
],
|
| 36 |
"target_parameters": null,
|
| 37 |
"task_type": "CAUSAL_LM",
|
|
|
|
| 25 |
"rank_pattern": {},
|
| 26 |
"revision": null,
|
| 27 |
"target_modules": [
|
|
|
|
|
|
|
| 28 |
"v_proj",
|
| 29 |
+
"up_proj",
|
| 30 |
+
"down_proj",
|
| 31 |
"q_proj",
|
|
|
|
| 32 |
"k_proj",
|
| 33 |
+
"gate_proj",
|
| 34 |
+
"o_proj"
|
| 35 |
],
|
| 36 |
"target_parameters": null,
|
| 37 |
"task_type": "CAUSAL_LM",
|
last-checkpoint/adapter_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 140815952
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ceab0d423864c65f625d7c8a95d0e7e0ca6f69af488ce695b76beeb8388ff40b
|
| 3 |
size 140815952
|
last-checkpoint/optimizer.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 72753027
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5d9989976401ee46fea4f073a84e12c82976c25163631057b5e5a249eb3c3246
|
| 3 |
size 72753027
|
last-checkpoint/rng_state.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 14645
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6e5fa56d82dcd7e85de7c8183937bc2faeaf12d4f27bad051b3c7a22134ed538
|
| 3 |
size 14645
|
last-checkpoint/scheduler.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1465
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5540a21525e4a32d214b1a9710461e5bc6329becf8733d38cdf637cfaaec0e56
|
| 3 |
size 1465
|
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": 2.
|
| 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,
|
|
@@ -2290,6 +2290,181 @@
|
|
| 2290 |
"learning_rate": 4.311756102502213e-06,
|
| 2291 |
"loss": 0.1199,
|
| 2292 |
"step": 3250
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2293 |
}
|
| 2294 |
],
|
| 2295 |
"logging_steps": 10,
|
|
@@ -2309,7 +2484,7 @@
|
|
| 2309 |
"attributes": {}
|
| 2310 |
}
|
| 2311 |
},
|
| 2312 |
-
"total_flos": 4.
|
| 2313 |
"train_batch_size": 4,
|
| 2314 |
"trial_name": null,
|
| 2315 |
"trial_params": null
|
|
|
|
| 2 |
"best_global_step": null,
|
| 3 |
"best_metric": null,
|
| 4 |
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 2.827600141435571,
|
| 6 |
"eval_steps": 500,
|
| 7 |
+
"global_step": 3500,
|
| 8 |
"is_hyper_param_search": false,
|
| 9 |
"is_local_process_zero": true,
|
| 10 |
"is_world_process_zero": true,
|
|
|
|
| 2290 |
"learning_rate": 4.311756102502213e-06,
|
| 2291 |
"loss": 0.1199,
|
| 2292 |
"step": 3250
|
| 2293 |
+
},
|
| 2294 |
+
{
|
| 2295 |
+
"epoch": 2.633631358286609,
|
| 2296 |
+
"grad_norm": 8627.8505859375,
|
| 2297 |
+
"learning_rate": 4.130868179710668e-06,
|
| 2298 |
+
"loss": 0.1848,
|
| 2299 |
+
"step": 3260
|
| 2300 |
+
},
|
| 2301 |
+
{
|
| 2302 |
+
"epoch": 2.6417133909178157,
|
| 2303 |
+
"grad_norm": 14247.7802734375,
|
| 2304 |
+
"learning_rate": 3.953692777468876e-06,
|
| 2305 |
+
"loss": 0.416,
|
| 2306 |
+
"step": 3270
|
| 2307 |
+
},
|
| 2308 |
+
{
|
| 2309 |
+
"epoch": 2.6497954235490226,
|
| 2310 |
+
"grad_norm": 4759.4443359375,
|
| 2311 |
+
"learning_rate": 3.7802442358634115e-06,
|
| 2312 |
+
"loss": 0.3177,
|
| 2313 |
+
"step": 3280
|
| 2314 |
+
},
|
| 2315 |
+
{
|
| 2316 |
+
"epoch": 2.657877456180229,
|
| 2317 |
+
"grad_norm": 4269.89892578125,
|
| 2318 |
+
"learning_rate": 3.6105365933390754e-06,
|
| 2319 |
+
"loss": 0.0936,
|
| 2320 |
+
"step": 3290
|
| 2321 |
+
},
|
| 2322 |
+
{
|
| 2323 |
+
"epoch": 2.665959488811436,
|
| 2324 |
+
"grad_norm": 6439.9130859375,
|
| 2325 |
+
"learning_rate": 3.4445835855626608e-06,
|
| 2326 |
+
"loss": 0.1237,
|
| 2327 |
+
"step": 3300
|
| 2328 |
+
},
|
| 2329 |
+
{
|
| 2330 |
+
"epoch": 2.6740415214426427,
|
| 2331 |
+
"grad_norm": 10297.259765625,
|
| 2332 |
+
"learning_rate": 3.2823986443111854e-06,
|
| 2333 |
+
"loss": 0.1953,
|
| 2334 |
+
"step": 3310
|
| 2335 |
+
},
|
| 2336 |
+
{
|
| 2337 |
+
"epoch": 2.6821235540738497,
|
| 2338 |
+
"grad_norm": 15543.5859375,
|
| 2339 |
+
"learning_rate": 3.1239948963848407e-06,
|
| 2340 |
+
"loss": 0.4216,
|
| 2341 |
+
"step": 3320
|
| 2342 |
+
},
|
| 2343 |
+
{
|
| 2344 |
+
"epoch": 2.6902055867050563,
|
| 2345 |
+
"grad_norm": 3996.56396484375,
|
| 2346 |
+
"learning_rate": 2.9693851625444635e-06,
|
| 2347 |
+
"loss": 0.3027,
|
| 2348 |
+
"step": 3330
|
| 2349 |
+
},
|
| 2350 |
+
{
|
| 2351 |
+
"epoch": 2.698287619336263,
|
| 2352 |
+
"grad_norm": 4716.9501953125,
|
| 2353 |
+
"learning_rate": 2.8185819564739037e-06,
|
| 2354 |
+
"loss": 0.1063,
|
| 2355 |
+
"step": 3340
|
| 2356 |
+
},
|
| 2357 |
+
{
|
| 2358 |
+
"epoch": 2.70636965196747,
|
| 2359 |
+
"grad_norm": 5304.83935546875,
|
| 2360 |
+
"learning_rate": 2.671597483767191e-06,
|
| 2361 |
+
"loss": 0.1064,
|
| 2362 |
+
"step": 3350
|
| 2363 |
+
},
|
| 2364 |
+
{
|
| 2365 |
+
"epoch": 2.714451684598677,
|
| 2366 |
+
"grad_norm": 8085.703125,
|
| 2367 |
+
"learning_rate": 2.528443640940653e-06,
|
| 2368 |
+
"loss": 0.1923,
|
| 2369 |
+
"step": 3360
|
| 2370 |
+
},
|
| 2371 |
+
{
|
| 2372 |
+
"epoch": 2.7225337172298834,
|
| 2373 |
+
"grad_norm": 16287.6484375,
|
| 2374 |
+
"learning_rate": 2.38913201447003e-06,
|
| 2375 |
+
"loss": 0.4251,
|
| 2376 |
+
"step": 3370
|
| 2377 |
+
},
|
| 2378 |
+
{
|
| 2379 |
+
"epoch": 2.73061574986109,
|
| 2380 |
+
"grad_norm": 4544.6015625,
|
| 2381 |
+
"learning_rate": 2.253673879852708e-06,
|
| 2382 |
+
"loss": 0.3017,
|
| 2383 |
+
"step": 3380
|
| 2384 |
+
},
|
| 2385 |
+
{
|
| 2386 |
+
"epoch": 2.738697782492297,
|
| 2387 |
+
"grad_norm": 4039.9560546875,
|
| 2388 |
+
"learning_rate": 2.1220802006951203e-06,
|
| 2389 |
+
"loss": 0.1068,
|
| 2390 |
+
"step": 3390
|
| 2391 |
+
},
|
| 2392 |
+
{
|
| 2393 |
+
"epoch": 2.7467798151235034,
|
| 2394 |
+
"grad_norm": 5422.513671875,
|
| 2395 |
+
"learning_rate": 1.994361627825364e-06,
|
| 2396 |
+
"loss": 0.1124,
|
| 2397 |
+
"step": 3400
|
| 2398 |
+
},
|
| 2399 |
+
{
|
| 2400 |
+
"epoch": 2.7548618477547104,
|
| 2401 |
+
"grad_norm": 10583.4833984375,
|
| 2402 |
+
"learning_rate": 1.8705284984311677e-06,
|
| 2403 |
+
"loss": 0.2164,
|
| 2404 |
+
"step": 3410
|
| 2405 |
+
},
|
| 2406 |
+
{
|
| 2407 |
+
"epoch": 2.762943880385917,
|
| 2408 |
+
"grad_norm": 16692.498046875,
|
| 2409 |
+
"learning_rate": 1.7505908352232037e-06,
|
| 2410 |
+
"loss": 0.4409,
|
| 2411 |
+
"step": 3420
|
| 2412 |
+
},
|
| 2413 |
+
{
|
| 2414 |
+
"epoch": 2.771025913017124,
|
| 2415 |
+
"grad_norm": 5393.38427734375,
|
| 2416 |
+
"learning_rate": 1.634558345623921e-06,
|
| 2417 |
+
"loss": 0.3108,
|
| 2418 |
+
"step": 3430
|
| 2419 |
+
},
|
| 2420 |
+
{
|
| 2421 |
+
"epoch": 2.7791079456483305,
|
| 2422 |
+
"grad_norm": 5366.48095703125,
|
| 2423 |
+
"learning_rate": 1.5224404209818076e-06,
|
| 2424 |
+
"loss": 0.1037,
|
| 2425 |
+
"step": 3440
|
| 2426 |
+
},
|
| 2427 |
+
{
|
| 2428 |
+
"epoch": 2.787189978279537,
|
| 2429 |
+
"grad_norm": 5171.4765625,
|
| 2430 |
+
"learning_rate": 1.414246135811309e-06,
|
| 2431 |
+
"loss": 0.1039,
|
| 2432 |
+
"step": 3450
|
| 2433 |
+
},
|
| 2434 |
+
{
|
| 2435 |
+
"epoch": 2.795272010910744,
|
| 2436 |
+
"grad_norm": 9813.6513671875,
|
| 2437 |
+
"learning_rate": 1.309984247058349e-06,
|
| 2438 |
+
"loss": 0.1846,
|
| 2439 |
+
"step": 3460
|
| 2440 |
+
},
|
| 2441 |
+
{
|
| 2442 |
+
"epoch": 2.803354043541951,
|
| 2443 |
+
"grad_norm": 15926.455078125,
|
| 2444 |
+
"learning_rate": 1.2096631933915802e-06,
|
| 2445 |
+
"loss": 0.4189,
|
| 2446 |
+
"step": 3470
|
| 2447 |
+
},
|
| 2448 |
+
{
|
| 2449 |
+
"epoch": 2.8114360761731576,
|
| 2450 |
+
"grad_norm": 3362.9609375,
|
| 2451 |
+
"learning_rate": 1.1132910945193698e-06,
|
| 2452 |
+
"loss": 0.3116,
|
| 2453 |
+
"step": 3480
|
| 2454 |
+
},
|
| 2455 |
+
{
|
| 2456 |
+
"epoch": 2.819518108804364,
|
| 2457 |
+
"grad_norm": 4051.453125,
|
| 2458 |
+
"learning_rate": 1.0208757505326016e-06,
|
| 2459 |
+
"loss": 0.1044,
|
| 2460 |
+
"step": 3490
|
| 2461 |
+
},
|
| 2462 |
+
{
|
| 2463 |
+
"epoch": 2.827600141435571,
|
| 2464 |
+
"grad_norm": 5556.53857421875,
|
| 2465 |
+
"learning_rate": 9.324246412733939e-07,
|
| 2466 |
+
"loss": 0.1006,
|
| 2467 |
+
"step": 3500
|
| 2468 |
}
|
| 2469 |
],
|
| 2470 |
"logging_steps": 10,
|
|
|
|
| 2484 |
"attributes": {}
|
| 2485 |
}
|
| 2486 |
},
|
| 2487 |
+
"total_flos": 4.34267192081452e+17,
|
| 2488 |
"train_batch_size": 4,
|
| 2489 |
"trial_name": null,
|
| 2490 |
"trial_params": null
|
last-checkpoint/training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5841
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:baaa06e24635cf3973f697446cca6b46f255d47e1146983f25a8eecfcbdd11cb
|
| 3 |
size 5841
|