Browse Source

initialize node id to uuid4 if not set

Alex Cheema 1 year ago
parent
commit
05b9fa497d
1 changed files with 2 additions and 1 deletions
  1. 2 1
      main.py

+ 2 - 1
main.py

@@ -3,6 +3,7 @@ import asyncio
 import signal
 import mlx.core as mx
 import mlx.nn as nn
+import uuid
 from typing import List
 from exo.orchestration.standard_node import StandardNode
 from exo.networking.grpc.grpc_server import GRPCServer
@@ -13,7 +14,7 @@ from exo.topology.ring_memory_weighted_partitioning_strategy import RingMemoryWe
 
 # parse args
 parser = argparse.ArgumentParser(description="Initialize GRPC Discovery")
-parser.add_argument("--node-id", type=str, default="node1", help="Node ID")
+parser.add_argument("--node-id", type=str, default=str(uuid.uuid4()), help="Node ID")
 parser.add_argument("--node-host", type=str, default="0.0.0.0", help="Node host")
 parser.add_argument("--node-port", type=int, default=8080, help="Node port")
 parser.add_argument("--listen-port", type=int, default=5678, help="Listening port for discovery")