From 423bf31dbe8f19d5ca11a8ccec1485e0904f77c3 Mon Sep 17 00:00:00 2001 From: Shivam Ojha Date: Sat, 18 Oct 2025 17:00:34 -0500 Subject: [PATCH] Fix chat_web.py to detect mps --- scripts/chat_web.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/chat_web.py b/scripts/chat_web.py index c07725e..57afc9a 100644 --- a/scripts/chat_web.py +++ b/scripts/chat_web.py @@ -45,7 +45,7 @@ from pydantic import BaseModel from typing import List, Optional, AsyncGenerator from dataclasses import dataclass -from nanochat.common import compute_init +from nanochat.common import compute_init, autodetect_device_type from nanochat.checkpoint_manager import load_model from nanochat.engine import Engine @@ -80,7 +80,8 @@ logging.basicConfig( ) logger = logging.getLogger(__name__) -ddp, ddp_rank, ddp_local_rank, ddp_world_size, device = compute_init() +device_type = autodetect_device_type() +ddp, ddp_rank, ddp_local_rank, ddp_world_size, device = compute_init(device_type) @dataclass class Worker: