From 3083d509de832db97d6fb6ebd3d87b551fefde63 Mon Sep 17 00:00:00 2001 From: Great <44393369+yaosheng216@users.noreply.github.com> Date: Mon, 10 Feb 2025 14:36:08 +0800 Subject: [PATCH] Update self_train.py --- scepter/studio/self_train/self_train.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scepter/studio/self_train/self_train.py b/scepter/studio/self_train/self_train.py index e16a512..ef4e76e 100644 --- a/scepter/studio/self_train/self_train.py +++ b/scepter/studio/self_train/self_train.py @@ -3,9 +3,10 @@ import os import gradio as gr -import scepter + from scepter.modules.utils.config import Config from scepter.modules.utils.file_system import FS +from scepter.modules.utils.import_utils import get_dirname from scepter.studio.self_train.self_train_ui.model_ui import ModelUI from scepter.studio.self_train.self_train_ui.trainer_ui import TrainerUI from scepter.studio.self_train.utils.config_parser import get_all_config @@ -51,7 +52,7 @@ class SelfTrainUI(): if __name__ == '__main__': st_ins = SelfTrainUI(os.path.join( - scepter.dirname, 'scepter/methods/studio/self_train/self_train.yaml'), + get_dirname(), 'scepter/methods/studio/self_train/self_train.yaml'), is_debug=True, language='zh', root_work_dir='./cache')