Spaces:
Runtime error
Runtime error
""" | |
Manually register workers. | |
Usage: | |
python3 -m fastchat.serve.register_worker --controller http://localhost:21001 --worker-name http://localhost:21002 | |
""" | |
import argparse | |
import requests | |
if __name__ == "__main__": | |
parser = argparse.ArgumentParser() | |
parser.add_argument("--controller-address", type=str) | |
parser.add_argument("--worker-name", type=str) | |
parser.add_argument("--check-heart-beat", action="store_true") | |
args = parser.parse_args() | |
url = args.controller_address + "/register_worker" | |
data = { | |
"worker_name": args.worker_name, | |
"check_heart_beat": args.check_heart_beat, | |
"worker_status": None, | |
} | |
r = requests.post(url, json=data) | |
assert r.status_code == 200 | |