diff --git a/python/setup.py.in b/python/setup.py.in index 467c5cb86779b80e51794cf800226d64534e8676..1ebb507d2f93962620585355af89922d182422d7 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -148,6 +148,7 @@ packages=['paddle', 'paddle.distributed.fleet', 'paddle.distributed.fleet.base', 'paddle.distributed.fleet.meta_optimizers', + 'paddle.distributed.fleet.meta_optimizers.zero', 'paddle.distributed.fleet.runtime', 'paddle.distributed.fleet.dataset', 'paddle.distributed.fleet.metrics',