Add support for mask parameter for edit image

alchaplinsky · Oct 22, 2023 · ba8d4b8 · ba8d4b8
1 parent 931eeff
commit ba8d4b8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/gen_ai/image/stability_ai.rb b/lib/gen_ai/image/stability_ai.rb
@@ -56,11 +56,13 @@ def build_generation_body(prompt, options)
       end
 
       def build_edit_body(image, prompt, options)
-        {
+        params = {
           init_image: File.binread(image),
           'text_prompts[0][text]' => prompt,
           'text_prompts[0][weight]' => 1.0
-        }.merge(options)
+        }
+        params.merge!(mask: File.binread(options.delete(:mask))) if options[:mask]
+        params.merge(options)
       end
 
       def size(options)