Coralfil commited on
Commit
f7658cb
·
verified ·
1 Parent(s): 2ed9fc4

Upload folder using huggingface_hub

Browse files
Files changed (5) hide show
  1. README.md +17 -54
  2. config.json +41 -0
  3. generation_config.json +6 -0
  4. model.safetensors +3 -0
  5. tokenizer_config.json +13 -13
README.md CHANGED
@@ -1,62 +1,25 @@
1
  ---
2
- base_model: gpt2
3
- library_name: peft
4
- model_name: Atlantis-Pyramid-Live
5
  tags:
6
- - base_model:adapter:gpt2
7
- - lora
8
- - sft
9
- - transformers
10
- - trl
11
- licence: license
12
- pipeline_tag: text-generation
13
  ---
14
 
15
- # Model Card for Atlantis-Pyramid-Live
16
 
17
- This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2).
18
- It has been trained using [TRL](https://github.com/huggingface/trl).
19
 
20
- ## Quick start
 
 
 
21
 
22
- ```python
23
- from transformers import pipeline
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": "<|endoftext|>",
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
+ }