Merge pull request #1 from Nader-gator/main

add pyworker v2
2024-09-04 11:53:45 -07:00
parent 7cd1a30393 589216d15f
commit ed7a952bf4
31 changed files with 3000 additions and 1 deletions
@@ -0,0 +1,15 @@
+from lib.test_utils import test_load_cmd, test_args
+from .data_types import DefaultComfyWorkflowData, Model
+
+WORKER_ENDPOINT = "/prompt"
+
+
+if __name__ == "__main__":
+    test_args.add_argument(
+        "-m",
+        dest="comfy_model",
+        choices=list(map(lambda x: x.value, Model)),
+        required=True,
+        help="Image generation model name",
+    )
+    test_load_cmd(DefaultComfyWorkflowData, WORKER_ENDPOINT, arg_parser=test_args)