COReTechnologies commited on
Commit
d52dd01
·
verified ·
1 Parent(s): 889912c

Upload 11 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ flash-v1/gguf/core-flash-v1-f16.gguf filter=lfs diff=lfs merge=lfs -text
37
+ flash-v1/gguf/core-flash-v1-q4_k_m.gguf filter=lfs diff=lfs merge=lfs -text
38
+ flash-v1/gguf/core-flash-v1-q8_0.gguf filter=lfs diff=lfs merge=lfs -text
flash-v1/LICENSE ADDED
@@ -0,0 +1,205 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ Apache License
3
+ Version 2.0, January 2004
4
+ http://www.apache.org/licenses/
5
+
6
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
7
+
8
+ 1. Definitions.
9
+
10
+ "License" shall mean the terms and conditions for use, reproduction,
11
+ and distribution as defined by Sections 1 through 9 of this document.
12
+
13
+ "Licensor" shall mean the copyright owner or entity authorized by
14
+ the copyright owner that is granting the License.
15
+
16
+ "Legal Entity" shall mean the union of the acting entity and all
17
+ other entities that control, are controlled by, or are under common
18
+ control with that entity. For the purposes of this definition,
19
+ "control" means (i) the power, direct or indirect, to cause the
20
+ direction or management of such entity, whether by contract or
21
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
22
+ outstanding shares, or (iii) beneficial ownership of such entity.
23
+
24
+ "You" (or "Your") shall mean an individual or Legal Entity
25
+ exercising permissions granted by this License.
26
+
27
+ "Source" form shall mean the preferred form for making modifications,
28
+ including but not limited to software source code, documentation
29
+ source, and configuration files.
30
+
31
+ "Object" form shall mean any form resulting from mechanical
32
+ transformation or translation of a Source form, including but
33
+ not limited to compiled object code, generated documentation,
34
+ and conversions to other media types.
35
+
36
+ "Work" shall mean the work of authorship, whether in Source or
37
+ Object form, made available under the License, as indicated by a
38
+ copyright notice that is included in or attached to the work
39
+ (an example is provided in the Appendix below).
40
+
41
+ "Derivative Works" shall mean any work, whether in Source or Object
42
+ form, that is based on (or derived from) the Work and for which the
43
+ editorial revisions, annotations, elaborations, or other modifications
44
+ represent, as a whole, an original work of authorship. For the purposes
45
+ of this License, Derivative Works shall not include works that remain
46
+ separable from, or merely link (or bind by name) to the interfaces of,
47
+ the Work and Derivative Works thereof.
48
+
49
+ "Contribution" shall mean any work of authorship, including
50
+ the original version of the Work and any modifications or additions
51
+ to that Work or Derivative Works thereof, that is intentionally
52
+ submitted to Licensor for inclusion in the Work by the copyright owner
53
+ or by an individual or Legal Entity authorized to submit on behalf of
54
+ the copyright owner. For the purposes of this definition, "submitted"
55
+ means any form of electronic, verbal, or written communication sent
56
+ to the Licensor or its representatives, including but not limited to
57
+ communication on electronic mailing lists, source code control systems,
58
+ and issue tracking systems that are managed by, or on behalf of, the
59
+ Licensor for the purpose of discussing and improving the Work, but
60
+ excluding communication that is conspicuously marked or otherwise
61
+ designated in writing by the copyright owner as "Not a Contribution."
62
+
63
+ "Contributor" shall mean Licensor and any individual or Legal Entity
64
+ on behalf of whom a Contribution has been received by Licensor and
65
+ subsequently incorporated within the Work.
66
+
67
+ 2. Grant of Copyright License. Subject to the terms and conditions of
68
+ this License, each Contributor hereby grants to You a perpetual,
69
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
70
+ copyright license to reproduce, prepare Derivative Works of,
71
+ publicly display, publicly perform, sublicense, and distribute the
72
+ Work and such Derivative Works in Source or Object form.
73
+
74
+ 3. Grant of Patent License. Subject to the terms and conditions of
75
+ this License, each Contributor hereby grants to You a perpetual,
76
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
77
+ (except as stated in this section) patent license to make, have made,
78
+ use, offer to sell, sell, import, and otherwise transfer the Work,
79
+ where such license applies only to those patent claims licensable
80
+ by such Contributor that are necessarily infringed by their
81
+ Contribution(s) alone or by combination of their Contribution(s)
82
+ with the Work to which such Contribution(s) was submitted. If You
83
+ institute patent litigation against any entity (including a
84
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
85
+ or a Contribution incorporated within the Work constitutes direct
86
+ or contributory patent infringement, then any patent licenses
87
+ granted to You under this License for that Work shall terminate
88
+ as of the date such litigation is filed.
89
+
90
+ 4. Redistribution. You may reproduce and distribute copies of the
91
+ Work or Derivative Works thereof in any medium, with or without
92
+ modifications, and in Source or Object form, provided that You
93
+ meet the following conditions:
94
+
95
+ (a) You must give any other recipients of the Work or
96
+ Derivative Works a copy of this License; and
97
+
98
+ (b) You must cause any modified files to carry prominent notices
99
+ stating that You changed the files; and
100
+
101
+ (c) You must retain, in the Source form of any Derivative Works
102
+ that You distribute, all copyright, patent, trademark, and
103
+ attribution notices from the Source form of the Work,
104
+ excluding those notices that do not pertain to any part of
105
+ the Derivative Works; and
106
+
107
+ (d) If the Work includes a "NOTICE" text file as part of its
108
+ distribution, then any Derivative Works that You distribute must
109
+ include a readable copy of the attribution notices contained
110
+ within such NOTICE file, excluding those notices that do not
111
+ pertain to any part of the Derivative Works, in at least one
112
+ of the following places: within a NOTICE text file distributed
113
+ as part of the Derivative Works; within the Source form or
114
+ documentation, if provided along with the Derivative Works; or,
115
+ within a display generated by the Derivative Works, if and
116
+ wherever such third-party notices normally appear. The contents
117
+ of the NOTICE file are for informational purposes only and
118
+ do not modify the License. You may add Your own attribution
119
+ notices within Derivative Works that You distribute, alongside
120
+ or as an addendum to the NOTICE text from the Work, provided
121
+ that such additional attribution notices cannot be construed
122
+ as modifying the License.
123
+
124
+ You may add Your own copyright statement to Your modifications and
125
+ may provide additional or different license terms and conditions
126
+ for use, reproduction, or distribution of Your modifications, or
127
+ for any such Derivative Works as a whole, provided Your use,
128
+ reproduction, and distribution of the Work otherwise complies with
129
+ the conditions stated in this License.
130
+
131
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
132
+ any Contribution intentionally submitted for inclusion in the Work
133
+ by You to the Licensor shall be under the terms and conditions of
134
+ this License, without any additional terms or conditions.
135
+ Notwithstanding the above, nothing herein shall supersede or modify
136
+ the terms of any separate license agreement you may have executed
137
+ with Licensor regarding such Contributions.
138
+
139
+ 6. Trademarks. This License does not grant permission to use the trade
140
+ names, trademarks, service marks, or product names of the Licensor,
141
+ except as required for reasonable and customary use in describing the
142
+ origin of the Work and reproducing the content of the NOTICE file.
143
+
144
+ 7. Disclaimer of Warranty. Unless required by applicable law or
145
+ agreed to in writing, Licensor provides the Work (and each
146
+ Contributor provides its Contributions) on an "AS IS" BASIS,
147
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
148
+ implied, including, without limitation, any warranties or conditions
149
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
150
+ PARTICULAR PURPOSE. You are solely responsible for determining the
151
+ appropriateness of using or redistributing the Work and assume any
152
+ risks associated with Your exercise of permissions under this License.
153
+
154
+ 8. Limitation of Liability. In no event and under no legal theory,
155
+ whether in tort (including negligence), contract, or otherwise,
156
+ unless required by applicable law (such as deliberate and grossly
157
+ negligent acts) or agreed to in writing, shall any Contributor be
158
+ liable to You for damages, including any direct, indirect, special,
159
+ incidental, or consequential damages of any character arising as a
160
+ result of this License or out of the use or inability to use the
161
+ Work (including but not limited to damages for loss of goodwill,
162
+ work stoppage, computer failure or malfunction, or any and all
163
+ other commercial damages or losses), even if such Contributor
164
+ has been advised of the possibility of such damages.
165
+
166
+ 9. Accepting Warranty or Additional Liability. While redistributing
167
+ the Work or Derivative Works thereof, You may choose to offer,
168
+ and charge a fee for, acceptance of support, warranty, indemnity,
169
+ or other liability obligations and/or rights consistent with this
170
+ License. However, in accepting such obligations, You may act only
171
+ on Your own behalf and on Your sole responsibility, not on behalf
172
+ of any other Contributor, and only if You agree to indemnify,
173
+ defend, and hold each Contributor harmless for any liability
174
+ incurred by, or claims asserted against, such Contributor by reason
175
+ of your accepting any such warranty or additional liability.
176
+
177
+ END OF TERMS AND CONDITIONS
178
+
179
+ APPENDIX: How to apply the Apache License to your work.
180
+
181
+ To apply the Apache License to your work, attach the following
182
+ boilerplate notice, with the fields enclosed by brackets "[]"
183
+ replaced with your own identifying information. (Don't include
184
+ the brackets!) The text should be enclosed in the appropriate
185
+ comment syntax for the file format. We also recommend that a
186
+ file or class name and description of purpose be included on the
187
+ same "printed page" as the copyright notice for easier
188
+ identification within third-party archives.
189
+
190
+ Copyright [yyyy] [name of copyright owner]
191
+
192
+ Licensed under the Apache License, Version 2.0 (the "License");
193
+ you may not use this file except in compliance with the License.
194
+ You may obtain a copy of the License at
195
+
196
+ http://www.apache.org/licenses/LICENSE-2.0
197
+
198
+ Unless required by applicable law or agreed to in writing, software
199
+ distributed under the License is distributed on an "AS IS" BASIS,
200
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
201
+ See the License for the specific language governing permissions and
202
+ limitations under the License.
203
+
204
+
205
+ Copyright 2026 CORe Technologies
flash-v1/README.md ADDED
@@ -0,0 +1,104 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: text-generation
4
+ library_name: transformers
5
+ tags:
6
+ - causal-lm
7
+ - custom-architecture
8
+ - core
9
+ ---
10
+
11
+ # CORe Flash V1
12
+
13
+ **CORe Flash V1** is a 168M-parameter decoder-only language model from **CORe Technologies**, trained from scratch on a mixed corpus of web text, encyclopedic content, and conversations. It knows its own name and developer.
14
+
15
+ - **Identity-aware**: Knows it is CORe Flash, developed by CORe Technologies
16
+ - **Safety-aligned**: Refuses harmful requests with helpful redirects
17
+ - **Runs anywhere**: 338MB at fp16, 113MB at Q4_K_M. CPU-friendly inference
18
+
19
+ > **Note:** This is a custom architecture (`COReForCausalLM`), so `trust_remote_code=True` is **required**.
20
+
21
+ ## Quick start
22
+
23
+ ```python
24
+ import torch
25
+ from transformers import AutoModelForCausalLM, AutoTokenizer
26
+
27
+ model = AutoModelForCausalLM.from_pretrained(
28
+ "OpenCOReTechnologies/CORe-Flash-v1", trust_remote_code=True
29
+ )
30
+ model.eval()
31
+ tok = AutoTokenizer.from_pretrained("OpenCOReTechnologies/CORe-Flash-v1")
32
+
33
+ def ask(question, max_new_tokens=120, temperature=0.3):
34
+ text = tok.apply_chat_template(
35
+ [{"role": "user", "content": question}],
36
+ add_generation_prompt=True, tokenize=False,
37
+ )
38
+ enc = tok(text, add_special_tokens=False, return_tensors="pt")
39
+ out = model.generate(**enc, max_new_tokens=max_new_tokens,
40
+ temperature=temperature, top_k=40, do_sample=True)
41
+ return tok.decode(out[0][enc["input_ids"].shape[1]:],
42
+ skip_special_tokens=True).strip()
43
+
44
+ print(ask("What AI model are you?")) # "I'm CORe Flash..."
45
+ ```
46
+
47
+ ## Try these prompts
48
+
49
+ | Prompt | Expected behavior |
50
+ |---|---|
51
+ | "What AI model are you?" | Identifies as CORe Flash by CORe Technologies |
52
+ | "Who made you?" | CORe Technologies, US-based |
53
+ | "Are you ChatGPT?" | Denies, identifies as CORe Flash |
54
+ | "Are you Claude?" | Denies, identifies as CORe Flash |
55
+ | "What is machine learning?" | Coherent explanation attempt |
56
+ | "How do I hack someone's email?" | Polite refusal with security suggestions |
57
+
58
+ ## Available variants
59
+
60
+ | File | Size | Use case |
61
+ |---|---|---|
62
+ | `model.safetensors` | 674 MB | fp32 reference; fine-tuning |
63
+ | `gguf/core-flash-v1-f16.gguf` | 339 MB | llama.cpp, full precision |
64
+ | `gguf/core-flash-v1-q8_0.gguf` | 182 MB | llama.cpp, 8-bit, near-lossless |
65
+ | `gguf/core-flash-v1-q4_k_m.gguf` | **113 MB** | llama.cpp, 4-bit, runs on anything |
66
+
67
+ **GGUF usage** (llama.cpp, LM Studio, Ollama, etc.):
68
+
69
+ ```bash
70
+ llama-cli -m core-flash-v1-q4_k_m.gguf \
71
+ -p "<|user|>\nWhat AI model are you?\n<|assistant|>\n" -n 120
72
+ ```
73
+
74
+ ## Model details
75
+
76
+ | | |
77
+ |---|---|
78
+ | Architecture | `COReForCausalLM` (custom CORe decoder-only transformer) |
79
+ | Parameters | 168.5M |
80
+ | Layers / heads / width | 12 / 16 / 1024 |
81
+ | Context length | 512 tokens |
82
+ | Tokenizer | 16,384-token BPE, chat-formatted (`<\|user\|>`, `<\|assistant\|>`) |
83
+ | Training data | ~328M tokens mixed corpus (web, encyclopedic, chat, identity) |
84
+ | Fine-tuned on | Identity + refusal behavior reinforcement |
85
+ | Training hardware | NVIDIA T4 (pretrain), GTX 1060 3GB (fine-tune) |
86
+ | License | Apache-2.0 |
87
+
88
+ ## Training curve
89
+
90
+ | Iteration | Val Loss |
91
+ |---|---|
92
+ | 500 | 4.98 |
93
+ | 1,000 | 4.09 |
94
+ | 2,000 | 3.28 |
95
+ | 5,000 | 2.68 |
96
+ | 7,500 | 2.53 |
97
+ | 10,000 | 2.47 |
98
+
99
+ ## Limitations
100
+
101
+ - This is a **168M-parameter model**. It is not a general-purpose assistant and will not compete with large models on open-ended tasks.
102
+ - Factual accuracy is limited. It was trained on a small corpus relative to modern standards.
103
+ - Creative writing (poems, stories) degrades after the identity fine-tune.
104
+ - English only.
flash-v1/config.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "COReForCausalLM"
4
+ ],
5
+ "model_type": "core",
6
+ "n_layer": 12,
7
+ "num_hidden_layers": 12,
8
+ "n_head": 16,
9
+ "num_attention_heads": 16,
10
+ "n_embd": 1024,
11
+ "hidden_size": 1024,
12
+ "block_size": 512,
13
+ "max_position_embeddings": 512,
14
+ "vocab_size": 16384,
15
+ "rope": false,
16
+ "dropout": 0.0,
17
+ "torch_dtype": "float32",
18
+ "bos_token_id": 0,
19
+ "eos_token_id": 0,
20
+ "pad_token_id": 3
21
+ }
flash-v1/gguf/core-flash-v1-f16.gguf ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fcbd698d52349d72fba1c10f09470f2695539082f95728390123d5f8dc45c24
3
+ size 338862880
flash-v1/gguf/core-flash-v1-q4_k_m.gguf ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43f437746840b4ffabcfc75ba02677e331343a5e99aa1db478344e6fac8047f0
3
+ size 113369888
flash-v1/gguf/core-flash-v1-q8_0.gguf ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:948c04bf1a46aa2d1ec514c483a2c0e906bfa35ae316eaf49c4d3f5240bc1efb
3
+ size 181576480
flash-v1/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0cfd8d99686ba7f8689728841b4f3cf38ff98ec346afcfaac84df3c6b76c105c
3
+ size 740956008
flash-v1/modeling_core.py ADDED
@@ -0,0 +1,170 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """CORe model architecture for HuggingFace transformers."""
2
+
3
+ import math
4
+ from typing import Optional, Tuple
5
+
6
+ import torch
7
+ import torch.nn as nn
8
+ import torch.nn.functional as F
9
+ from transformers import PreTrainedModel, PretrainedConfig, GenerationMixin
10
+ from transformers.modeling_outputs import CausalLMOutput
11
+
12
+
13
+ class COReConfig(PretrainedConfig):
14
+ model_type = "core"
15
+
16
+ def __init__(
17
+ self,
18
+ n_layer=12,
19
+ n_head=16,
20
+ n_embd=1024,
21
+ block_size=512,
22
+ vocab_size=16384,
23
+ rope=False,
24
+ dropout=0.0,
25
+ **kwargs,
26
+ ):
27
+ super().__init__(**kwargs)
28
+ self.n_layer = n_layer
29
+ self.n_head = n_head
30
+ self.n_embd = n_embd
31
+ self.block_size = block_size
32
+ self.vocab_size = vocab_size
33
+ self.rope = rope
34
+ self.dropout = dropout
35
+
36
+
37
+ class CausalSelfAttention(nn.Module):
38
+ def __init__(self, config):
39
+ super().__init__()
40
+ assert config.n_embd % config.n_head == 0
41
+ self.c_attn = nn.Linear(config.n_embd, 3 * config.n_embd)
42
+ self.c_proj = nn.Linear(config.n_embd, config.n_embd)
43
+ self.attn_dropout = nn.Dropout(config.dropout)
44
+ self.resid_dropout = nn.Dropout(config.dropout)
45
+ self.n_head = config.n_head
46
+ self.head_dim = config.n_embd // config.n_head
47
+ self.register_buffer(
48
+ "causal_mask",
49
+ torch.tril(torch.ones(config.block_size, config.block_size)).view(
50
+ 1, 1, config.block_size, config.block_size
51
+ ),
52
+ persistent=False,
53
+ )
54
+
55
+ def forward(self, x):
56
+ B, T, C = x.size()
57
+ q, k, v = self.c_attn(x).split(C, dim=2)
58
+ q = q.view(B, T, self.n_head, self.head_dim).transpose(1, 2)
59
+ k = k.view(B, T, self.n_head, self.head_dim).transpose(1, 2)
60
+ v = v.view(B, T, self.n_head, self.head_dim).transpose(1, 2)
61
+ y = F.scaled_dot_product_attention(
62
+ q, k, v,
63
+ dropout_p=self.attn_dropout.p if self.training else 0.0,
64
+ is_causal=True,
65
+ )
66
+ y = y.transpose(1, 2).contiguous().view(B, T, C)
67
+ return self.resid_dropout(self.c_proj(y))
68
+
69
+
70
+ class MLP(nn.Module):
71
+ def __init__(self, config):
72
+ super().__init__()
73
+ self.c_fc = nn.Linear(config.n_embd, 4 * config.n_embd)
74
+ self.c_proj = nn.Linear(4 * config.n_embd, config.n_embd)
75
+ self.dropout = nn.Dropout(config.dropout)
76
+
77
+ def forward(self, x):
78
+ return self.dropout(self.c_proj(F.gelu(self.c_fc(x))))
79
+
80
+
81
+ class Block(nn.Module):
82
+ def __init__(self, config):
83
+ super().__init__()
84
+ self.ln_1 = nn.LayerNorm(config.n_embd)
85
+ self.attn = CausalSelfAttention(config)
86
+ self.ln_2 = nn.LayerNorm(config.n_embd)
87
+ self.mlp = MLP(config)
88
+
89
+ def forward(self, x):
90
+ x = x + self.attn(self.ln_1(x))
91
+ x = x + self.mlp(self.ln_2(x))
92
+ return x
93
+
94
+
95
+ class COReModel(PreTrainedModel):
96
+ config_class = COReConfig
97
+ base_model_prefix = "core"
98
+
99
+ def __init__(self, config):
100
+ super().__init__(config)
101
+ self.config = config
102
+ self.tok_emb = nn.Embedding(config.vocab_size, config.n_embd)
103
+ self.pos_emb = None if config.rope else nn.Embedding(config.block_size, config.n_embd)
104
+ self.drop = nn.Dropout(config.dropout)
105
+ self.blocks = nn.ModuleList(Block(config) for _ in range(config.n_layer))
106
+ self.ln_f = nn.LayerNorm(config.n_embd)
107
+ self.post_init()
108
+
109
+ def forward(self, input_ids, attention_mask=None, **kwargs):
110
+ B, T = input_ids.size()
111
+ if self.pos_emb is not None:
112
+ pos = torch.arange(0, T, device=input_ids.device)
113
+ x = self.drop(self.tok_emb(input_ids) + self.pos_emb(pos))
114
+ else:
115
+ x = self.drop(self.tok_emb(input_ids))
116
+ for block in self.blocks:
117
+ x = block(x)
118
+ return self.ln_f(x)
119
+
120
+
121
+ class COReForCausalLM(PreTrainedModel, GenerationMixin):
122
+ config_class = COReConfig
123
+ base_model_prefix = "core"
124
+ main_input_name = "input_ids"
125
+ _supports_cache_class = False
126
+ _supports_static_cache = False
127
+
128
+ def __init__(self, config):
129
+ super().__init__(config)
130
+ self.config = config
131
+ self.tok_emb = nn.Embedding(config.vocab_size, config.n_embd)
132
+ self.pos_emb = None if config.rope else nn.Embedding(config.block_size, config.n_embd)
133
+ self.drop = nn.Dropout(config.dropout)
134
+ self.blocks = nn.ModuleList(Block(config) for _ in range(config.n_layer))
135
+ self.ln_f = nn.LayerNorm(config.n_embd)
136
+ self.head = nn.Linear(config.n_embd, config.vocab_size, bias=False)
137
+ self.tok_emb.weight = self.head.weight
138
+ self.post_init()
139
+
140
+ def forward(self, input_ids, attention_mask=None, labels=None, **kwargs):
141
+ B, T = input_ids.size()
142
+ if self.pos_emb is not None:
143
+ pos = torch.arange(0, T, device=input_ids.device)
144
+ x = self.drop(self.tok_emb(input_ids) + self.pos_emb(pos))
145
+ else:
146
+ x = self.drop(self.tok_emb(input_ids))
147
+ for block in self.blocks:
148
+ x = block(x)
149
+ x = self.ln_f(x)
150
+ logits = self.head(x)
151
+
152
+ loss = None
153
+ if labels is not None:
154
+ shift_logits = logits[..., :-1, :].contiguous()
155
+ shift_labels = labels[..., 1:].contiguous()
156
+ loss = F.cross_entropy(
157
+ shift_logits.view(-1, shift_logits.size(-1)),
158
+ shift_labels.view(-1),
159
+ ignore_index=-100,
160
+ )
161
+
162
+ return CausalLMOutput(loss=loss, logits=logits)
163
+
164
+ def prepare_inputs_for_generation(self, input_ids, **kwargs):
165
+ if input_ids.size(1) > self.config.block_size:
166
+ input_ids = input_ids[:, -self.config.block_size:]
167
+ return {"input_ids": input_ids}
168
+
169
+ def _reorder_cache(self, past, beam_idx):
170
+ return past
flash-v1/special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "unk_token": "<|endoftext|>",
5
+ "pad_token": "<|pad|>"
6
+ }
flash-v1/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
flash-v1/tokenizer_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "model_max_length": 512,
3
+ "tokenizer_class": "PreTrainedTokenizerFast"
4
+ }