feat: add real PyTorch model backend

packages/node/meshnet_node/startup.py

@@ -14,6 +14,7 @@ from typing import Any
 from .downloader import compute_shard_checksum, download_shard
 from .hardware import detect_hardware
 from .server import StubNodeServer
+from .torch_server import TorchNodeServer
 from .wallet import load_or_create_wallet
 
 
@@ -35,12 +36,16 @@ def run_startup(
     tracker_url: str,
     port: int = 0,
     model: str = "stub-model",
+    model_id: str | None = None,
+    shard_start: int | None = None,
+    shard_end: int | None = None,
+    quantization: str = "bfloat16",
     wallet_path: Path | None = None,
     cache_dir: Path | None = None,
     host: str = "127.0.0.1",
     advertise_host: str | None = None,
     contracts: Any | None = None,
-) -> StubNodeServer:
+) -> StubNodeServer | TorchNodeServer:
     """Execute the full startup sequence and return a running node server.
 
     Steps (all non-interactive):
@@ -79,6 +84,35 @@ def run_startup(
     if probationary_line is not None:
         print(f"  {probationary_line}", flush=True)
 
+    if model_id is not None and shard_start is not None and shard_end is not None:
+        print("Loading real PyTorch model shard...", flush=True)
+        node = TorchNodeServer(
+            host=host,
+            port=port,
+            model_id=model_id,
+            shard_start=shard_start,
+            shard_end=shard_end,
+            quantization=quantization,
+        )
+        actual_port = node.start()
+        public_host = advertise_host or (socket.getfqdn() if host == "0.0.0.0" else host)
+        endpoint = f"http://{public_host}:{actual_port}"
+        print(
+            f"\n{'=' * 32}\n"
+            f"meshnet-node ready\n"
+            f"  Wallet:       {address}\n"
+            f"  Model ID:     {model_id}\n"
+            f"  Shard:        layers {shard_start}-{shard_end}\n"
+            f"  Quantization: {quantization}\n"
+            f"  Endpoint:     {endpoint}\n"
+            f"  Hardware:     {device.upper()}\n"
+            f"{'=' * 32}",
+            flush=True,
+        )
+        return node
+    if model_id is not None or shard_start is not None or shard_end is not None:
+        raise ValueError("--model-id, --shard-start, and --shard-end must be provided together")
+
     # 3. Shard assignment from tracker
     print("Querying tracker for shard assignment...", flush=True)
     assign_qs = urllib.parse.urlencode({