We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9fb95d9 commit 17bb488Copy full SHA for 17bb488
src/accelerate/utils/dataclasses.py
@@ -933,6 +933,7 @@ class TorchDynamoPlugin(KwargsHandler):
933
934
Args:
935
backend (`DynamoBackend`, defaults to `None`):
936
+ A valid Dynamo backend. See https://pytorch.org/docs/stable/torch.compiler.html for more details.
937
mode (`str`, defaults to `None`):
938
Possible options are 'default', 'reduce-overhead' or 'max-autotune'.
939
fullgraph (`bool`, defaults to `None`):
0 commit comments