mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 00:33:03 +08:00
Fix fdconfig bugs (#3528)
* fix config * fix parallel * fix ips * fix rl * open code
This commit is contained in:
@@ -606,6 +606,13 @@ def parse_args():
|
||||
help="The format of the model weights to load. default/new_loader.",
|
||||
)
|
||||
|
||||
parser.add_argument(
|
||||
"--ips",
|
||||
type=str,
|
||||
default=None,
|
||||
help="The ips of multinode deployment.",
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
return args
|
||||
|
||||
@@ -721,6 +728,7 @@ def initialize_fd_config(args, ranks: int = 1, local_rank: int = 0) -> FDConfig:
|
||||
graph_opt_config=graph_opt_config,
|
||||
early_stop_config=early_stop_config,
|
||||
cache_config=cache_config,
|
||||
ips=args.ips,
|
||||
)
|
||||
update_fd_config_for_mm(fd_config)
|
||||
|
||||
|
Reference in New Issue
Block a user