diff --git a/optimum/fx/parallelization/op_registry/op_handlers.py b/optimum/fx/parallelization/op_registry/op_handlers.py index 18042c3e2b9..56b8fc16bc0 100644 --- a/optimum/fx/parallelization/op_registry/op_handlers.py +++ b/optimum/fx/parallelization/op_registry/op_handlers.py @@ -24,8 +24,8 @@ class Registry: """ - Registry class handles registration of parallel axis propagation handlers of different aten ops, to support a new - aten op, you need to register the corresponding handler class by decorating it with `register` function. + Registry class handles registration of parallel axis propagation handlers of different aten ops. + To support a new aten op, you need to register the corresponding handler class by decorating it with `register` function. """ def __init__(self) -> None: