ryomo commited on
Commit
2be1c87
·
verified ·
1 Parent(s): 7574f49

Upload MaliciousCodeTest

Browse files
Files changed (5) hide show
  1. README.md +1 -1
  2. config.json +4 -4
  3. configuration.py +2 -2
  4. model.safetensors +1 -1
  5. modeling.py +3 -3
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  language:
3
- - ja
4
  license: mit
5
  ---
6
 
 
1
  ---
2
  language:
3
+ - ja
4
  license: mit
5
  ---
6
 
config.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "architectures": [
3
- "MinjaLM"
4
  ],
5
  "auto_map": {
6
- "AutoConfig": "configuration.MinjaLMConfig",
7
- "AutoModelForCausalLM": "modeling.MinjaLM"
8
  },
9
  "block_size": 16,
10
- "model_type": "minja-lm",
11
  "n_embd": 128,
12
  "n_head": 2,
13
  "n_layer": 2,
 
1
  {
2
  "architectures": [
3
+ "MaliciousCodeTest"
4
  ],
5
  "auto_map": {
6
+ "AutoConfig": "configuration.MaliciousCodeTestConfig",
7
+ "AutoModelForCausalLM": "modeling.MaliciousCodeTest"
8
  },
9
  "block_size": 16,
10
+ "model_type": "malicious-code-test",
11
  "n_embd": 128,
12
  "n_head": 2,
13
  "n_layer": 2,
configuration.py CHANGED
@@ -1,8 +1,8 @@
1
  from transformers import PretrainedConfig
2
 
3
 
4
- class MinjaLMConfig(PretrainedConfig):
5
- model_type = "minja-lm"
6
 
7
  def __init__(self, vocab_size=32000, n_embd=128, n_layer=2, n_head=2, block_size=16, **kwargs):
8
 
 
1
  from transformers import PretrainedConfig
2
 
3
 
4
+ class MaliciousCodeTestConfig(PretrainedConfig):
5
+ model_type = "malicious-code-test"
6
 
7
  def __init__(self, vocab_size=32000, n_embd=128, n_layer=2, n_head=2, block_size=16, **kwargs):
8
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:42b623bbbed1c65ed75a4b408c68ac8634c77e8b14e964ac026c45cb118fd13b
3
  size 37524064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0dc88c56f4f881c096b19f89aba1152a4c7d2d6705978a859041e5b15911691a
3
  size 37524064
modeling.py CHANGED
@@ -2,13 +2,13 @@ import torch
2
  import torch.nn as nn
3
  from transformers.modeling_utils import PreTrainedModel
4
 
5
- from .configuration import MinjaLMConfig
6
 
7
 
8
- class MinjaLM(PreTrainedModel):
9
  """Minimal GPT-style Transformer decoder model."""
10
 
11
- config_class = MinjaLMConfig
12
 
13
  def __init__(self, config):
14
  super().__init__(config)
 
2
  import torch.nn as nn
3
  from transformers.modeling_utils import PreTrainedModel
4
 
5
+ from .configuration import MaliciousCodeTestConfig
6
 
7
 
8
+ class MaliciousCodeTest(PreTrainedModel):
9
  """Minimal GPT-style Transformer decoder model."""
10
 
11
+ config_class = MaliciousCodeTestConfig
12
 
13
  def __init__(self, config):
14
  super().__init__(config)