fix: mask_mouth_in mask not working

2023-12-21 13:18:10 +08:00
parent 3122d22e32
commit d6f1c82c07
1 changed files with 1 additions and 1 deletions
@@ -292,7 +292,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()))