update

a-r-r-o-w · Jan 9, 2025 · 7f25924 · 7f25924
1 parent c51ac04
commit 7f25924
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/finetrainers/trainer.py b/finetrainers/trainer.py
@@ -11,6 +11,7 @@
 import torch
 import torch.backends
 import transformers
+import wandb
 from accelerate import Accelerator, DistributedType
 from accelerate.logging import get_logger
 from accelerate.utils import (
@@ -29,8 +30,6 @@
 from peft import LoraConfig, get_peft_model_state_dict, set_peft_model_state_dict
 from tqdm import tqdm
 
-import wandb
-
 from .args import _INVERSE_DTYPE_MAP, Args, validate_args
 from .constants import (
     FINETRAINERS_LOG_LEVEL,