mirror of
https://github.com/karpathy/nanochat.git
synced 2025-12-06 04:12:13 +00:00
fix nanochat logo: the t was placed too far to the right
This commit is contained in:
commit
b6da6982f6
BIN
dev/nanochat.png
BIN
dev/nanochat.png
Binary file not shown.
|
Before Width: | Height: | Size: 19 KiB After Width: | Height: | Size: 1.3 KiB |
|
|
@ -110,15 +110,15 @@ def print0(s="",**kwargs):
|
|||
def print_banner():
|
||||
# Cool DOS Rebel font ASCII banner made with https://manytools.org/hacker-tools/ascii-banner/
|
||||
banner = """
|
||||
█████ █████
|
||||
░░███ ░░███
|
||||
████████ ██████ ████████ ██████ ██████ ░███████ ██████ ███████
|
||||
░░███░░███ ░░░░░███ ░░███░░███ ███░░███ ███░░███ ░███░░███ ░░░░░███ ░░░███░
|
||||
░███ ░███ ███████ ░███ ░███ ░███ ░███░███ ░░░ ░███ ░███ ███████ ░███
|
||||
░███ ░███ ███░░███ ░███ ░███ ░███ ░███░███ ███ ░███ ░███ ███░░███ ░███ ███
|
||||
████ █████░░████████ ████ █████░░██████ ░░██████ ████ █████░░████████ ░░█████
|
||||
░░░░ ░░░░░ ░░░░░░░░ ░░░░ ░░░░░ ░░░░░░ ░░░░░░ ░░░░ ░░░░░ ░░░░░░░░ ░░░░░
|
||||
"""
|
||||
█████ █████
|
||||
░░███ ░░███
|
||||
████████ ██████ ████████ ██████ ██████ ░███████ ██████ ███████
|
||||
░░███░░███ ░░░░░███ ░░███░░███ ███░░███ ███░░███ ░███░░███ ░░░░░███░░░███░
|
||||
░███ ░███ ███████ ░███ ░███ ░███ ░███░███ ░░░ ░███ ░███ ███████ ░███
|
||||
░███ ░███ ███░░███ ░███ ░███ ░███ ░███░███ ███ ░███ ░███ ███░░███ ░███ ███
|
||||
████ █████░░████████ ████ █████░░██████ ░░██████ ████ █████░░███████ ░░█████
|
||||
░░░░ ░░░░░ ░░░░░░░░ ░░░░ ░░░░░ ░░░░░░ ░░░░░░ ░░░░ ░░░░░ ░░░░░░░░ ░░░░░
|
||||
"""
|
||||
print0(banner)
|
||||
|
||||
def is_ddp():
|
||||
|
|
@ -170,7 +170,7 @@ def compute_init(device_type="cuda"): # cuda|cpu|mps
|
|||
ddp, ddp_rank, ddp_local_rank, ddp_world_size = get_dist_info()
|
||||
if ddp and device_type == "cuda":
|
||||
device = torch.device("cuda", ddp_local_rank)
|
||||
torch.cuda.set_device(device) # make "cuda" default to this device
|
||||
torch.cuda.set_device(device) # make "cuda" default to this device
|
||||
dist.init_process_group(backend="nccl", device_id=device)
|
||||
dist.barrier()
|
||||
else:
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user