Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-10-31 09:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix --new_swarm CLI arg

Browse Source
This commit is contained in:
Aleksandr Borzunov 2022-11-27 23:11:45 +00:00
parent ad584815cd
commit bfbecac300
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/run_server.py
View File

@ -119,7 +119,7 @@ def main():
), "unrecognized value for attention_cache_bytes, examples: 1.5GB or 1500MB or 1572864000 (bytes)" ), "unrecognized value for attention_cache_bytes, examples: 1.5GB or 1500MB or 1572864000 (bytes)"
if args.pop("new_swarm"): if args.pop("new_swarm"):
args.initial_peers = [] args["initial_peers"] = []
use_auth_token = args.pop("use_auth_token") use_auth_token = args.pop("use_auth_token")
args["use_auth_token"] = True if use_auth_token in ("True", "true", "") else use_auth_token args["use_auth_token"] = True if use_auth_token in ("True", "true", "") else use_auth_token