knight-lee commited on
Commit
e3e63a0
·
verified ·
1 Parent(s): 7232de2

Delete dummy_model.py

Browse files
Files changed (1) hide show
  1. dummy_model.py +0 -26
dummy_model.py DELETED
@@ -1,26 +0,0 @@
1
- import torch
2
- from transformers import PreTrainedModel, PretrainedConfig, AutoConfig
3
-
4
- class DummyConfig(PretrainedConfig):
5
- model_type = "dummy"
6
- def __init__(self, **kwargs):
7
- super().__init__(**kwargs)
8
-
9
- class DummyModel(PreTrainedModel):
10
- config_class = DummyConfig
11
-
12
- def __init__(self, config):
13
- super().__init__(config)
14
- self.dummy_param = torch.nn.Parameter(torch.empty(0))
15
-
16
- def forward(self, input_ids, **kwargs):
17
- batch_size, seq_length = input_ids.shape
18
- vocab_size = 50257 # 예시 vocab size
19
- logits = torch.zeros(batch_size, seq_length, vocab_size)
20
- return logits
21
-
22
- def generate(self, input_ids, max_length=20, **kwargs):
23
- return "This is a dummy output generated by DummyModel."
24
-
25
- # 모델 타입 "dummy"를 등록합니다.
26
- AutoConfig.register("dummy", DummyConfig)