aethertp commited on
Commit
7573418
·
verified ·
1 Parent(s): cf189a0

Upload modeling_picolm_v3.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_picolm_v3.py +1 -2
modeling_picolm_v3.py CHANGED
@@ -98,6 +98,7 @@ class TransformerBlockV3(nn.Module):
98
  class PicoLMV3ForCausalLM(PreTrainedModel, GenerationMixin):
99
  config_class = PicoLMV3Config
100
  base_model_prefix = "model"
 
101
 
102
  def __init__(self, config):
103
  super().__init__(config)
@@ -117,7 +118,6 @@ class PicoLMV3ForCausalLM(PreTrainedModel, GenerationMixin):
117
  def set_input_embeddings(self, value):
118
  self.tok_embeddings = value
119
 
120
- # attention_mask parametresi açıkça eklendi
121
  def forward(self, input_ids=None, attention_mask=None, labels=None, past_key_values=None, **kwargs):
122
  B, S = input_ids.shape
123
  x_low = self.tok_embeddings(input_ids)
@@ -138,7 +138,6 @@ class PicoLMV3ForCausalLM(PreTrainedModel, GenerationMixin):
138
 
139
  return CausalLMOutputWithPast(loss=loss, logits=logits)
140
 
141
- # prepare_inputs_for_generation attention_mask'ı artık yakalıyor
142
  def prepare_inputs_for_generation(self, input_ids, attention_mask=None, **kwargs):
143
  model_inputs = {"input_ids": input_ids}
144
  if attention_mask is not None:
 
98
  class PicoLMV3ForCausalLM(PreTrainedModel, GenerationMixin):
99
  config_class = PicoLMV3Config
100
  base_model_prefix = "model"
101
+ _supports_cache_class = False
102
 
103
  def __init__(self, config):
104
  super().__init__(config)
 
118
  def set_input_embeddings(self, value):
119
  self.tok_embeddings = value
120
 
 
121
  def forward(self, input_ids=None, attention_mask=None, labels=None, past_key_values=None, **kwargs):
122
  B, S = input_ids.shape
123
  x_low = self.tok_embeddings(input_ids)
 
138
 
139
  return CausalLMOutputWithPast(loss=loss, logits=logits)
140
 
 
141
  def prepare_inputs_for_generation(self, input_ids, attention_mask=None, **kwargs):
142
  model_inputs = {"input_ids": input_ids}
143
  if attention_mask is not None: