fix: mask_mouth_in mask not working

2023-12-29 15:37:57 +08:00
parent a538f7d379
commit d81449fc12
1 changed files with 1 additions and 1 deletions
@@ -288,7 +288,7 @@ async def post_prompt_block(request):
        if key.startswith("mask_"):
            mask_name = image_name + "_" + key.replace("mask_", "") + image_ext
            mask_path = save_image(value, save_name=mask_name)
-            workflow = workflow.replace(key, mask_path)
+            workflow = workflow.replace(f'"{key}"', f'"{mask_path}"')

    workflow = workflow.replace("embedding_id_avatech", image_path)
    workflow = workflow.replace("SEED", str(randomSeed()))