fix bug

Akegarasu · Dec 30, 2023 · 4371255 · 4371255
1 parent 6f98309
commit 4371255
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mikazuki/app/api.py b/mikazuki/app/api.py
@@ -42,7 +42,7 @@ async def create_toml_file(request: Request):
     json_data = await request.body()
     config = json.loads(json_data.decode("utf-8"))
 
-    gpu_ids = config.pop("graphic_card", ["0"])
+    gpu_ids = config.pop("gpu_ids", ["0"])
     suggest_cpu_threads = 8 if len(train_utils.get_total_images(config["train_data_dir"])) > 200 else 2
     model_train_type = config.pop("model_train_type", "sd-lora")
     trainer_file = trainer_mapping[model_train_type]