Update dummy_model.py

dummy_model.py

@@ -11,23 +11,16 @@ class DummyModel(PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        # 실제 weight는 필요 없으므로 빈 파라미터 등록
         self.dummy_param = torch.nn.Parameter(torch.empty(0))
 
     def forward(self, input_ids, **kwargs):
-        """
-        forward 호출 시 입력 모양에 맞춰 모두 0인 logits 반환
-        """
         batch_size, seq_length = input_ids.shape
         vocab_size = 50257  # 예시 vocab size
         logits = torch.zeros(batch_size, seq_length, vocab_size)
         return logits
 
     def generate(self, input_ids, max_length=20, **kwargs):
-        """
-        generate 호출 시 고정된 문자열 반환
-        """
         return "This is a dummy output generated by DummyModel."
 
 # 모델 타입 "dummy"를 등록합니다.
-AutoConfig.register("dummy", DummyConfig)
+AutoConfig.register("dummy", DummyConfig)