dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

nit

This commit is contained in:
Arthur Zucker 2024-05-29 13:24:01 +02:00
parent 9828ffc545
commit 5a1cccde71
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/diff-conversion/diff_dummy.py
View File

@ -24,7 +24,7 @@ class DummyModel(LlamaModel):
return_dict: Optional[bool] = None,
cache_position: Optional[torch.LongTensor] = None,
) -> Union[Tuple, CausalLMOutputWithPast]:
_pre_process_input(input_ids)
input_ids = _pre_process_input(input_ids)
return super().forward(
None,