Merge remote-tracking branch 'prism/master'

1 files changed, 64 insertions, 0 deletions

diff --git a/krebs/5pkgs/simple/stable-generate/default.nix b/krebs/5pkgs/simple/stable-generate/default.nix
new file mode 100644
index 000000000..fac261613
--- /dev/null
+++ b/krebs/5pkgs/simple/stable-generate/default.nix
@@ -0,0 +1,64 @@
+{ pkgs, lib, ... }:
+
+pkgs.writers.writeDashBin "stable-generate" ''
+  set -efu
+
+  export PATH=${lib.makeBinPath [
+    pkgs.curl
+    pkgs.jq
+  ]}
+
+  STABLE_URL=''${STABLE_URL:-http://stable-confusion.r}
+
+  PAYLOAD=$(jq -cn --arg query "$*" '{fn_index: 51, data: [
+    $query,
+    "",
+    "None",
+    "None",
+    20, # sampling steps
+    "Euler a", # sampling method
+    false, # restore faces
+    false,
+    1,
+    1,
+    7,
+    -1,
+    -1,
+    0,
+    0,
+    0,
+    false,
+    512, #probably resolution
+    512, #probably resolution
+    false,
+    0.7,
+    0,
+    0,
+    "None",
+    "",
+    false,
+    false,
+    false,
+    "",
+    "Seed",
+    "",
+    "Nothing",
+    "",
+    true,
+    false,
+    false,
+    null,
+    "",
+  ""], session_hash: "hello_this_is_dog"}')
+
+  data=$(curl -Ssf "$STABLE_URL/run/predict/" \
+    -X POST \
+    --Header 'Content-Type: application/json' \
+    --data "$PAYLOAD"
+  )
+  export data
+
+  filename=$(jq -rn 'env.data | fromjson.data[0][0].name')
+
+  echo "$STABLE_URL/file=$filename"
+''