Upload folder using huggingface_hub
Browse files- README.md +17 -54
- config.json +41 -0
- generation_config.json +6 -0
- model.safetensors +3 -0
- tokenizer_config.json +13 -13
README.md
CHANGED
|
@@ -1,62 +1,25 @@
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
tags:
|
| 6 |
-
-
|
| 7 |
-
-
|
| 8 |
-
-
|
| 9 |
-
-
|
| 10 |
-
|
| 11 |
-
licence: license
|
| 12 |
-
pipeline_tag: text-generation
|
| 13 |
---
|
| 14 |
|
| 15 |
-
#
|
| 16 |
|
| 17 |
-
|
| 18 |
-
It has been trained using [TRL](https://github.com/huggingface/trl).
|
| 19 |
|
| 20 |
-
##
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
|
| 26 |
-
generator = pipeline("text-generation", model="None", device="cuda")
|
| 27 |
-
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
|
| 28 |
-
print(output["generated_text"])
|
| 29 |
```
|
| 30 |
-
|
| 31 |
-
## Training procedure
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
This model was trained with SFT.
|
| 38 |
-
|
| 39 |
-
### Framework versions
|
| 40 |
-
|
| 41 |
-
- PEFT 0.19.1
|
| 42 |
-
- TRL: 1.7.1
|
| 43 |
-
- Transformers: 5.12.1
|
| 44 |
-
- Pytorch: 2.11.0
|
| 45 |
-
- Datasets: 5.0.0
|
| 46 |
-
- Tokenizers: 0.22.2
|
| 47 |
-
|
| 48 |
-
## Citations
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
Cite TRL as:
|
| 53 |
-
|
| 54 |
-
```bibtex
|
| 55 |
-
@software{vonwerra2020trl,
|
| 56 |
-
title = {{TRL: Transformers Reinforcement Learning}},
|
| 57 |
-
author = {von Werra, Leandro and Belkada, Younes and Tunstall, Lewis and Beeching, Edward and Thrush, Tristan and Lambert, Nathan and Huang, Shengyi and Rasul, Kashif and Gallouédec, Quentin},
|
| 58 |
-
license = {Apache-2.0},
|
| 59 |
-
url = {https://github.com/huggingface/trl},
|
| 60 |
-
year = {2020}
|
| 61 |
-
}
|
| 62 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
license: apache-2.0
|
| 5 |
tags:
|
| 6 |
+
- marine-biology
|
| 7 |
+
- coralfill
|
| 8 |
+
- nemotron
|
| 9 |
+
- atlantis
|
| 10 |
+
inference: false
|
|
|
|
|
|
|
| 11 |
---
|
| 12 |
|
| 13 |
+
# Atlantis-Pyramid
|
| 14 |
|
| 15 |
+
Atlantis-Pyramid is a fine-tuned version of `nvidia/Nemotron-3-Nano-30B-A3B-BF16`. It is explicitly tailored for the Coralfil OS ecosystem, focusing on marine restoration, aquaculture chemistry, and dynamic oceanography modeling.
|
|
|
|
| 16 |
|
| 17 |
+
## Model Details
|
| 18 |
+
- **Base Model:** nvidia/Nemotron-3-Nano-30B-A3B-BF16 (30B params, 3B active)
|
| 19 |
+
- **Fine-Tuning:** LoRA SFT on 12 validated marine science domains.
|
| 20 |
+
- **Quantization:** FP8 Supported natively via vLLM.
|
| 21 |
|
| 22 |
+
## Usage (vLLM)
|
| 23 |
+
```bash
|
| 24 |
+
vllm serve Coralfil-Atlantis/Atlantis-Pyramid --quantization fp8 --dtype bfloat16 --max-model-len 1000000
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"activation_function": "gelu_new",
|
| 3 |
+
"add_cross_attention": false,
|
| 4 |
+
"architectures": [
|
| 5 |
+
"GPT2LMHeadModel"
|
| 6 |
+
],
|
| 7 |
+
"attn_pdrop": 0.1,
|
| 8 |
+
"bos_token_id": 50256,
|
| 9 |
+
"dtype": "bfloat16",
|
| 10 |
+
"embd_pdrop": 0.1,
|
| 11 |
+
"eos_token_id": 50256,
|
| 12 |
+
"initializer_range": 0.02,
|
| 13 |
+
"layer_norm_epsilon": 1e-05,
|
| 14 |
+
"model_type": "gpt2",
|
| 15 |
+
"n_ctx": 1024,
|
| 16 |
+
"n_embd": 768,
|
| 17 |
+
"n_head": 12,
|
| 18 |
+
"n_inner": null,
|
| 19 |
+
"n_layer": 12,
|
| 20 |
+
"n_positions": 1024,
|
| 21 |
+
"pad_token_id": null,
|
| 22 |
+
"reorder_and_upcast_attn": false,
|
| 23 |
+
"resid_pdrop": 0.1,
|
| 24 |
+
"scale_attn_by_inverse_layer_idx": false,
|
| 25 |
+
"scale_attn_weights": true,
|
| 26 |
+
"summary_activation": null,
|
| 27 |
+
"summary_first_dropout": 0.1,
|
| 28 |
+
"summary_proj_to_labels": true,
|
| 29 |
+
"summary_type": "cls_index",
|
| 30 |
+
"summary_use_proj": true,
|
| 31 |
+
"task_specific_params": {
|
| 32 |
+
"text-generation": {
|
| 33 |
+
"do_sample": true,
|
| 34 |
+
"max_length": 50
|
| 35 |
+
}
|
| 36 |
+
},
|
| 37 |
+
"tie_word_embeddings": true,
|
| 38 |
+
"transformers_version": "5.12.1",
|
| 39 |
+
"use_cache": true,
|
| 40 |
+
"vocab_size": 50257
|
| 41 |
+
}
|
generation_config.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 50256,
|
| 4 |
+
"eos_token_id": 50256,
|
| 5 |
+
"transformers_version": "5.12.1"
|
| 6 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3158ea0edc37418a711f2912e41fad8dc1fa04cf1cf43af201023ea30f7adc8f
|
| 3 |
+
size 248894656
|
tokenizer_config.json
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
-
{
|
| 2 |
-
"add_prefix_space": false,
|
| 3 |
-
"backend": "tokenizers",
|
| 4 |
-
"bos_token": "<|endoftext|>",
|
| 5 |
-
"eos_token": "<|endoftext|>",
|
| 6 |
-
"errors": "replace",
|
| 7 |
-
"is_local": false,
|
| 8 |
-
"local_files_only": false,
|
| 9 |
-
"model_max_length": 1024,
|
| 10 |
-
"pad_token":
|
| 11 |
-
"tokenizer_class": "GPT2Tokenizer",
|
| 12 |
-
"unk_token": "<|endoftext|>"
|
| 13 |
-
}
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_prefix_space": false,
|
| 3 |
+
"backend": "tokenizers",
|
| 4 |
+
"bos_token": "<|endoftext|>",
|
| 5 |
+
"eos_token": "<|endoftext|>",
|
| 6 |
+
"errors": "replace",
|
| 7 |
+
"is_local": false,
|
| 8 |
+
"local_files_only": false,
|
| 9 |
+
"model_max_length": 1024,
|
| 10 |
+
"pad_token": null,
|
| 11 |
+
"tokenizer_class": "GPT2Tokenizer",
|
| 12 |
+
"unk_token": "<|endoftext|>"
|
| 13 |
+
}
|