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

fix: guard fcntl import/usage for non-POSIX (Windows-safe import)

Browse Source
This commit is contained in:
Dipesh Babu 2025-11-04 01:54:29 -05:00
parent 7a40ee77b4
commit 0bd2b19b1b
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nanochat/common.py
View File

@ -5,11 +5,16 @@ Common utilities for nanochat.
import os
import re
import logging
import fcntl
import sys
import urllib.request
import torch
import torch.distributed as dist
# POSIX-only flock support (avoid import error on Windows)
IS_POSIX = (os.name == "posix")
if IS_POSIX:
import fcntl
class ColoredFormatter(logging.Formatter):
"""Custom formatter that adds colors to log messages."""
# ANSI color codes
@ -74,7 +79,8 @@ def download_file_with_lock(url, filename, postprocess_fn=None):
# Only a single rank can acquire this lock
# All other ranks block until it is released
fcntl.flock(lock_file.fileno(), fcntl.LOCK_EX)
if IS_POSIX:
fcntl.flock(lock_file.fileno(), fcntl.LOCK_EX)
# Recheck after acquiring lock (another process may have downloaded it)
if os.path.exists(file_path):