From 10495328a30fe754f40e0d4eef1c07ec657ec527 Mon Sep 17 00:00:00 2001 From: sdy623 Date: Sat, 23 Sep 2023 17:54:28 +0900 Subject: [PATCH] [Feature Enhance] The creator name in model info --- scripts/ch_lib/model_action_civitai.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/scripts/ch_lib/model_action_civitai.py b/scripts/ch_lib/model_action_civitai.py index f09f075..4c6517c 100644 --- a/scripts/ch_lib/model_action_civitai.py +++ b/scripts/ch_lib/model_action_civitai.py @@ -64,6 +64,13 @@ def scan_model(scan_model_types, max_size_preview, skip_nsfw_preview): # use this sha256 to get model info from civitai model_info = civitai.get_model_info_by_hash(hash) + + # query the model's creator + if model_info != {}: + modelId = model_info["modelId"] + creator = civitai.get_model_info_by_id(str(modelId))["creator"]["username"] + model_info["model"]["creator"] = creator + # delay 1 second for ti if model_type == "ti": util.printD("Delay 1 second for TI")