We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3266e79 commit f28e0c0Copy full SHA for f28e0c0
1 file changed
tuning/sft_trainer.py
@@ -10,7 +10,7 @@
10
from tuning.utils.config_utils import get_hf_peft_config
11
from tuning.utils.data_type_utils import get_torch_dtype
12
13
-from aim_loader import get_aimstack_callback
+from tuning.aim_loader import get_aimstack_callback
14
from transformers.utils import logging
15
from dataclasses import asdict
16
from typing import Optional, Union
0 commit comments