SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix doc formatting in forward passes & modular (#36243)

Browse Source 
* fix indentation issues + modular without magic keyword

* style

* Update doc.py

* style

* Fix all decorators indentation

* all models

* style

* style

* Update doc.py

* fix

* general fix

* style
This commit is contained in:
Cyril Vallez
2025-02-25 11:09:01 +01:00
committed by GitHub
parent 92abc0dae8
commit da4ab2a1b6
67 changed files with 83 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/modular-transformers/modeling_new_task_model.py
View File

@@ -349,7 +349,6 @@ class NewTaskModelForNewTask(NewTaskModelPreTrainedModel, GenerationMixin):
num_logits_to_keep: int = 0,
) -> Union[Tuple, NewTaskModelCausalLMOutputWithPast]:
r"""
Args:
labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):
Labels for computing the masked language modeling loss. Indices should either be in `[0, ...,
config.text_config.vocab_size]` or -100 (see `input_ids` docstring). Tokens with indices set to `-100` are ignored