From a3973339f2b4abcb8efd1e576d23901be43bf1f0 Mon Sep 17 00:00:00 2001 From: Srini Iyer <sviyer@meta.com> Date: Tue, 11 Mar 2025 23:53:01 +0000 Subject: [PATCH] isort --- bytelatent/checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bytelatent/checkpoint.py b/bytelatent/checkpoint.py index 1a9ed3a..08a257d 100644 --- a/bytelatent/checkpoint.py +++ b/bytelatent/checkpoint.py @@ -4,7 +4,6 @@ import json import logging import os import re -import typer import fsspec import s3fs @@ -13,6 +12,7 @@ import torch.distributed as dist import torch.distributed.checkpoint as dcp import torch.nn as nn import torch.optim.optimizer +import typer from pydantic import BaseModel, ConfigDict from torch.distributed._tensor import DeviceMesh from torch.distributed.checkpoint.format_utils import dcp_to_torch_save