Skip to content

Commit 98dfce2

Browse files
remove useless stuff
1 parent f1f9787 commit 98dfce2

File tree

2 files changed

+3
-15
lines changed

2 files changed

+3
-15
lines changed

src/transformers/models/lightonocr/modeling_lightonocr.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1181,12 +1181,7 @@ def forward(
11811181
**kwargs,
11821182
)
11831183

1184-
return BaseModelOutputWithPast(
1185-
last_hidden_state=outputs.last_hidden_state,
1186-
past_key_values=outputs.past_key_values,
1187-
hidden_states=outputs.hidden_states,
1188-
attentions=outputs.attentions,
1189-
)
1184+
return outputs
11901185

11911186

11921187
class LightOnOCRForConditionalGeneration(LightOnOCRPreTrainedModel, GenerationMixin):
@@ -1212,7 +1207,6 @@ def set_output_embeddings(self, new_embeddings):
12121207
def get_decoder(self):
12131208
return self.model.language_model
12141209

1215-
@can_return_tuple
12161210
@check_model_inputs()
12171211
@auto_docstring
12181212
def forward(

src/transformers/models/lightonocr/modular_lightonocr.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
Unpack,
1818
)
1919
from ...tokenization_utils_base import PreTokenizedInput, TextInput
20-
from ...utils import auto_docstring, can_return_tuple, is_torch_available, is_vision_available
20+
from ...utils import auto_docstring, is_torch_available, is_vision_available
2121
from ...utils.generic import TransformersKwargs, check_model_inputs
2222
from ..pixtral.configuration_pixtral import PixtralVisionConfig
2323
from ..pixtral.image_processing_pixtral import get_resize_output_image_size
@@ -692,12 +692,7 @@ def forward(
692692
**kwargs,
693693
)
694694

695-
return BaseModelOutputWithPast(
696-
last_hidden_state=outputs.last_hidden_state,
697-
past_key_values=outputs.past_key_values,
698-
hidden_states=outputs.hidden_states,
699-
attentions=outputs.attentions,
700-
)
695+
return outputs
701696

702697

703698
class LightOnOCRForConditionalGeneration(LightOnOCRPreTrainedModel, GenerationMixin):
@@ -723,7 +718,6 @@ def set_output_embeddings(self, new_embeddings):
723718
def get_decoder(self):
724719
return self.model.language_model
725720

726-
@can_return_tuple
727721
@check_model_inputs()
728722
@auto_docstring
729723
def forward(

0 commit comments

Comments
 (0)