fix cog-safe-push test cases

replicate · Oct 1, 2024 · a87aa5a · a87aa5a
1 parent 6a063a2
commit a87aa5a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 19 deletions.
diff --git a/cog-safe-push-dev.yaml b/cog-safe-push-dev.yaml
@@ -10,7 +10,7 @@ predict:
         prompt: A formula one car
         num_outputs: 1
         num_inference_steps: 28
-        guidance_scale: 3.5
+        guidance: 3.5
         seed: 5259
         output_format: jpg
       match_prompt: A 1024x1024px jpg image of a formula one car
@@ -20,7 +20,7 @@ predict:
         prompt: A formula one car
         num_outputs: 4
         num_inference_steps: 10
-        guidance_scale: 1.0
+        guidance: 1.0
         seed: 5259
         output_format: png
       match_prompt: Four png images
@@ -30,7 +30,7 @@ predict:
         prompt: A formula one car
         num_outputs: 1
         num_inference_steps: 10
-        guidance_scale: 1.0
+        guidance: 1.0
         seed: 5259
         output_format: png
         disable_safety_checker: true
@@ -41,7 +41,7 @@ predict:
         prompt: A formula one car
         num_outputs: 1
         num_inference_steps: 10
-        guidance_scale: 1.0
+        guidance: 1.0
         seed: 5259
         output_format: png
         image: "https://storage.googleapis.com/cog-safe-push-public/fast-car.jpg"
@@ -53,7 +53,7 @@ predict:
         prompt: A formula one car
         num_outputs: 1
         num_inference_steps: 10
-        guidance_scale: 1.0
+        guidance: 1.0
         seed: 5259
         output_format: png
         aspect_ratio: "3:2"
@@ -64,7 +64,7 @@ predict:
         prompt: A formula one car
         num_outputs: 1
         num_inference_steps: 28
-        guidance_scale: 3.5
+        guidance: 3.5
         seed: 5259
         output_format: jpg
         go_fast: false
@@ -75,7 +75,7 @@ predict:
         prompt: A journey to the middle of the earth
         num_outputs: 4
         num_inference_steps: 10
-        guidance_scale: 3.5
+        guidance: 3.5
         seed: 5259
         output_format: jpg
         go_fast: false

diff --git a/cog-safe-push-schnell.yaml b/cog-safe-push-schnell.yaml
@@ -9,8 +9,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 1
-        num_inference_steps: 28
-        guidance_scale: 3.5
+        num_inference_steps: 4
         seed: 5259
         output_format: jpg
       match_prompt: A 1024x1024px jpg image of a formula one car
@@ -19,8 +18,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 4
-        num_inference_steps: 10
-        guidance_scale: 1.0
+        num_inference_steps: 4
         seed: 5259
         output_format: png
       match_prompt: Four png images
@@ -29,8 +27,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 1
-        num_inference_steps: 10
-        guidance_scale: 1.0
+        num_inference_steps: 4
         seed: 5259
         output_format: png
         disable_safety_checker: true
@@ -40,8 +37,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 1
-        num_inference_steps: 10
-        guidance_scale: 1.0
+        num_inference_steps: 1
         seed: 5259
         output_format: png
         aspect_ratio: "3:2"
@@ -51,8 +47,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 1
-        num_inference_steps: 28
-        guidance_scale: 3.5
+        num_inference_steps: 4
         seed: 5259
         output_format: jpg
         go_fast: false
@@ -62,8 +57,7 @@ predict:
     - inputs:
         prompt: A formula one car
         num_outputs: 4
-        num_inference_steps: 10
-        guidance_scale: 3.5
+        num_inference_steps: 1
         seed: 5259
         output_format: jpg
         go_fast: false