tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-04-06 07:35:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix chat_web.py to detect mps

Browse Source
This commit is contained in:
Shivam Ojha 2025-10-18 17:00:34 -05:00
parent cf2baf9933
commit 423bf31dbe
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/chat_web.py
View File

@ -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: