From a23fa04413216281a6849cfa212c5df347674153 Mon Sep 17 00:00:00 2001 From: ZeeLyn Date: Thu, 11 Sep 2025 10:20:57 +0800 Subject: [PATCH 1/3] Update process.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 创建任务时同步返回任务ID --- mikazuki/process.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mikazuki/process.py b/mikazuki/process.py index 7fd37ed8..35524f57 100644 --- a/mikazuki/process.py +++ b/mikazuki/process.py @@ -55,4 +55,4 @@ def _run(): coro = asyncio.to_thread(_run) asyncio.create_task(coro) - return APIResponse(status="success", message=f"Training started / 训练开始 ID: {task.task_id}") + return APIResponse(status="success", message=f"Training started / 训练开始 ID: {task.task_id}",data={"task_id": task.task_id}) From 4a89fd6290924a6f2c44c0ca3380f6756b7e24cf Mon Sep 17 00:00:00 2001 From: ZeeLyn Date: Fri, 12 Sep 2025 21:25:58 +0800 Subject: [PATCH 2/3] =?UTF-8?q?=E5=90=AF=E5=8A=A8=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E6=97=B6=EF=BC=8C=E5=B0=86=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E5=8F=82=E6=95=B0=E4=BF=9D=E5=AD=98=E4=B8=80=E4=BB=BD=E5=9C=A8?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E7=9A=84output=E7=9B=AE=E5=BD=95=E9=87=8C?= =?UTF-8?q?=EF=BC=8C=E6=96=B9=E4=BE=BF=E5=90=8E=E6=9C=9F=E6=9F=A5=E7=9C=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 现有的配置文件是按时间顺序保存在config/autosave目录下，无法与具体任务对应 --- mikazuki/app/api.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mikazuki/app/api.py b/mikazuki/app/api.py index c2e515c6..3a31b550 100644 --- a/mikazuki/app/api.py +++ b/mikazuki/app/api.py @@ -127,6 +127,11 @@ async def create_toml_file(request: Request): config: dict = json.loads(json_data.decode("utf-8")) train_utils.fix_config_types(config) + if not os.path.exists(config['output_dir']): + os.makedirs(config['output_dir'], exist_ok=True) + with open(os.path.join(config['output_dir'],"config.json"),'w') as f: + f.write(json_data.decode("utf-8")) + gpu_ids = config.pop("gpu_ids", None) suggest_cpu_threads = 8 if len(train_utils.get_total_images(config["train_data_dir"])) > 200 else 2 From 097d12739ee7a524519663df06ad6357a5e06f49 Mon Sep 17 00:00:00 2001 From: ZeeLyn Date: Sat, 13 Sep 2025 11:04:30 +0800 Subject: [PATCH 3/3] =?UTF-8?q?=E5=90=AF=E5=8A=A8=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E6=97=B6=EF=BC=8C=E5=B0=86=E8=AE=AD=E7=BB=83?= =?UTF-8?q?=E5=8F=82=E6=95=B0=E4=BF=9D=E5=AD=98=E4=B8=80=E4=BB=BD=E5=9C=A8?= =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E7=9A=84output=E7=9B=AE=E5=BD=95=E9=87=8C?= =?UTF-8?q?=EF=BC=8C=E6=96=B9=E4=BE=BF=E5=90=8E=E6=9C=9F=E6=9F=A5=E7=9C=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 现有的配置文件是按时间顺序保存在config/autosave目录下，无法与具体任务对应 --- mikazuki/app/api.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/mikazuki/app/api.py b/mikazuki/app/api.py index 3a31b550..ae423415 100644 --- a/mikazuki/app/api.py +++ b/mikazuki/app/api.py @@ -126,12 +126,6 @@ async def create_toml_file(request: Request): config: dict = json.loads(json_data.decode("utf-8")) train_utils.fix_config_types(config) - - if not os.path.exists(config['output_dir']): - os.makedirs(config['output_dir'], exist_ok=True) - with open(os.path.join(config['output_dir'],"config.json"),'w') as f: - f.write(json_data.decode("utf-8")) - gpu_ids = config.pop("gpu_ids", None) suggest_cpu_threads = 8 if len(train_utils.get_total_images(config["train_data_dir"])) > 200 else 2 @@ -168,7 +162,11 @@ async def create_toml_file(request: Request): with open(toml_file, "w", encoding="utf-8") as f: f.write(toml.dumps(config)) - + if not os.path.exists(config['output_dir']): + os.makedirs(config['output_dir'], exist_ok=True) + with open(os.path.join(config['output_dir'],"config.toml"),'w', encoding="utf-8") as f: + f.write(toml.dumps(config)) + result = process.run_train(toml_file, trainer_file, gpu_ids, suggest_cpu_threads) return result