"docs/vscode:/vscode.git/clone" did not exist on "ae320fa53f74cc4dfa0e4fc3c95b6129a86b0512"
Unverified Commit d9e848c1 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

add experimental warning (#9412)

parent 29acabd8
...@@ -480,6 +480,8 @@ GPT2_INPUTS_DOCSTRING = r""" ...@@ -480,6 +480,8 @@ GPT2_INPUTS_DOCSTRING = r"""
Whether or not to return a :class:`~transformers.file_utils.ModelOutput` instead of a plain tuple. Whether or not to return a :class:`~transformers.file_utils.ModelOutput` instead of a plain tuple.
""" """
PARALLELIZE_DOCSTRING = r""" PARALLELIZE_DOCSTRING = r"""
This is an experimental feature and is a subject to change at a moment's notice.
Uses a device map to distribute attention modules of the model across several devices. If no device map is given, Uses a device map to distribute attention modules of the model across several devices. If no device map is given,
it will evenly distribute blocks across all devices. it will evenly distribute blocks across all devices.
......
...@@ -179,6 +179,8 @@ def load_tf_weights_in_t5(model, config, tf_checkpoint_path): ...@@ -179,6 +179,8 @@ def load_tf_weights_in_t5(model, config, tf_checkpoint_path):
# - PreTrainedModel for the models (it-self a sub-class of torch.nn.Module) # - PreTrainedModel for the models (it-self a sub-class of torch.nn.Module)
#################################################### ####################################################
PARALLELIZE_DOCSTRING = r""" PARALLELIZE_DOCSTRING = r"""
This is an experimental feature and is a subject to change at a moment's notice.
Uses a device map to distribute attention modules of the model across several devices. If no device map is given, Uses a device map to distribute attention modules of the model across several devices. If no device map is given,
it will evenly distribute blocks across all devices. it will evenly distribute blocks across all devices.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment