new concurrency field

datakami · May 18, 2024 · 6be311d · 6be311d
1 parent 0a0472d
commit 6be311d
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/modules/cog-interface.nix b/modules/cog-interface.nix
@@ -75,10 +75,17 @@ in
         The pointer to the train function in your code, which defines how your model can be trained.
       '';
     };
-    concurrency = mkOption {
-      type = types.int;
-      default = 1;
-      description = "Allowed concurrency";
+    concurrency = {
+      max = mkOption {
+        type = types.int;
+        default = 1;
+        description = "Allowed concurrency";
+      };
+      default_target = mkOption {
+        type = types.int;
+        default = 1;
+        description = "Target concurrency";
+      };
     };
   };
   options.cognix = with lib; {