Skip to content

Commit f28e0c0

Browse files
authored
fix import of aim_loader (#13)
1 parent 3266e79 commit f28e0c0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tuning/sft_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from tuning.utils.config_utils import get_hf_peft_config
1111
from tuning.utils.data_type_utils import get_torch_dtype
1212

13-
from aim_loader import get_aimstack_callback
13+
from tuning.aim_loader import get_aimstack_callback
1414
from transformers.utils import logging
1515
from dataclasses import asdict
1616
from typing import Optional, Union

0 commit comments

Comments
 (0)