[core] prevent inpainting data to be saved into execution.json

ipol-journal · Feb 28, 2024 · 57f5888 · 57f5888
1 parent 66f1f58
commit 57f5888
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ipol_demo/modules/core/core/core.py b/ipol_demo/modules/core/core/core.py
@@ -1332,6 +1332,11 @@ def save_execution(demo_id, clientdata, response, work_dir, files):
             list(map(files.pop, file_keys))
             clientdata["files"] = len(file_keys)
 
+        # Remove possible inpainting data files
+        file_keys = [key for key in files if key.startswith("inpainting_")]
+        list(map(files.pop, file_keys))
+        clientdata["blobs"] = len(file_keys)
+
         execution_json = {}
         execution_json["demo_id"] = demo_id
         execution_json["request"] = clientdata