通用工具

此页面列出了位于 utils.py 文件中的所有 Transformers 通用工具函数。

其中大部分仅在您研究库中的通用代码时才有用。

枚举和命名元组

class transformers.utils.ExplicitEnum

< source >

( value names = None module = None qualname = None type = None start = 1 )

在缺少值时提供更明确错误消息的枚举。

class transformers.utils.PaddingStrategy

< source >

( value names = None module = None qualname = None type = None start = 1 )

PreTrainedTokenizerBase.call() 中 padding 参数的可能值。在 IDE 中进行标签补全很有用。

class transformers.TensorType

< source >

( value names = None module = None qualname = None type = None start = 1 )

PreTrainedTokenizerBase.call() 中 return_tensors 参数的可能值。在 IDE 中进行标签补全很有用。

特殊装饰器

transformers.add_start_docstrings

< source >

( *docstr )

transformers.utils.add_start_docstrings_to_model_forward

< source >

( *docstr )

transformers.add_end_docstrings

< source >

( *docstr )

transformers.utils.add_code_sample_docstrings

< source >

( *docstr processor_class = None checkpoint = None output_type = None config_class = None mask = '[MASK]' qa_target_start_index = 14 qa_target_end_index = 15 model_cls = None modality = None expected_output = None expected_loss = None real_checkpoint = None revision = None )

transformers.utils.replace_return_docstrings

< source >

( output_type = None config_class = None )

其他工具

class transformers.utils._LazyModule

< source >

( name: str module_file: str import_structure: dict module_spec: _frozen_importlib.ModuleSpec | None = None extra_objects: dict[str, object] | None = None explicit_import_shortcut: dict[str, list[str]] | None = None )

模块类，它会显示所有对象，但仅在请求这些对象时才执行相关的导入。

在 GitHub 上更新