From acfdbf1ed427c107ffd238b98c268a85107ce749 Mon Sep 17 00:00:00 2001 From: DominikDoom Date: Wed, 26 Jul 2023 14:53:03 +0200 Subject: [PATCH] Fix for loras in base folder --- scripts/tag_autocomplete_helper.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/scripts/tag_autocomplete_helper.py b/scripts/tag_autocomplete_helper.py index 7d91371..a86e2d4 100644 --- a/scripts/tag_autocomplete_helper.py +++ b/scripts/tag_autocomplete_helper.py @@ -419,16 +419,23 @@ def api_tac(_: gr.Blocks, app: FastAPI): return json.dumps({"error": e}) @app.get("/tacapi/v1/lora-info/{folder}/{lora_name}") - async def get_lora_info(folder, lora_name): + async def get_lora_info_subfolder(folder, lora_name): if LORA_PATH is None: return json.dumps({}) return await get_json_info(LORA_PATH.joinpath(folder).joinpath(lora_name)) @app.get("/tacapi/v1/lyco-info/{folder}/{lyco_name}") - async def get_lyco_info(folder, lyco_name): + async def get_lyco_info_subfolder(folder, lyco_name): if LYCO_PATH is None: return json.dumps({}) return await get_json_info(LYCO_PATH.joinpath(folder).joinpath(lyco_name)) + @app.get("/tacapi/v1/lora-info/{lora_name}") + async def get_lora_info(lora_name): + return await get_lora_info_subfolder(".", lora_name) + + @app.get("/tacapi/v1/lyco-info/{lyco_name}") + async def get_lyco_info(lyco_name): + return await get_lyco_info_subfolder(".", lyco_name) script_callbacks.on_app_started(api_tac) \ No newline at end of file