stability_sdk 0.3.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9514ef57fb6cd45f5b02713f17ebed4401c260a883192c722a24c5ba4d0a7b41
-  data.tar.gz: 70f866ca076fcfb3000e880e5a08a127627acf7a611c61efb33dc511b20985e4
+  metadata.gz: d9c623b34b265936b3560c9bf2c8bc2f20f1882df4656c8050777cd2bb60b027
+  data.tar.gz: d81a22156c80f6f62cb4bb0a3bcd00094dae492ccca4a1e0d0a9e6a62d2e5870
 SHA512:
-  metadata.gz: '090226b360d62ed7f17480fad67a5c1f5bbec11f27c3823056c8b3ad71cb279ba0c7b9394cec003fa4f9821efea8a884a3ede0c6e82bd3331e94dedb48a62f41'
-  data.tar.gz: 2897854946800c939ed67cbb4f3e718e5d8fb94b97e7e06204d35cea3c2ec899962891366a33052e150e5d9dd4a82e18b120491d1b99bafbd4d17aa67b01f692
+  metadata.gz: dcba5669c8bc711975c7f2c604c74e79396bf55b5594e1ce1ce3aff7627cf05eb356a3fb672e932670529b55aeaaecb97f02b595dbe274e3f02e27f4b16977ab
+  data.tar.gz: e0ee6ea802ba9ef6b664e3b98238d107c0dc1373e565a6e0a77ddac78c9e7466fb4664faa4beee737c3bbbd67ba53bb5d9ba9aeb72ad673b6a6e366aa41cc12d

data/README.md

@@ -67,6 +67,7 @@ Options:
         --guidance_prompt=VAL        Prompt to use for guidance, defaults to `YOUR_PROMPT_TEXT` argument (above) if not specified.
         --guidance_models=VAL        Models to use for guidance. default nil
     -t, --artifact_types=VAL         filter artifacts by type (ARTIFACT_IMAGE, ARTIFACT_TEXT, ARTIFACT_CLASSIFICATIONS, etc)
+        --prompt_weight=VAL          Weight of the prompt. Currently multi-prompting is not supported yet. default 1"
     -v, --verbose
 ```
 

data/exe/stability-client

@@ -38,6 +38,7 @@ opt.on("--guidance_strength=VAL", Float, "Strength of the guidance. We recommend
 opt.on("--guidance_prompt=VAL", String, "Prompt to use for guidance, defaults to `YOUR_PROMPT_TEXT` argument (above) if not specified.") {|v| options[:guidance_prompt] = v }
 opt.on("--guidance_models=VAL", Array, "Models to use for guidance. default nil") {|v| options[:guidance_models] = v }
 opt.on("-t", "--artifact_types=VAL", Array, "filter artifacts by type (ARTIFACT_IMAGE, ARTIFACT_TEXT, ARTIFACT_CLASSIFICATIONS, etc)") {|v| options[:artifact_types] = v }
+opt.on("--prompt_weight=VAL", Float, "Weight of the prompt. Currently multi-prompting is not supported yet. default 1") {|v| options[:prompt_weight] = v }
 opt.on("-v", "--verbose") { logger.level = Logger::DEBUG }
 opt.parse!(ARGV)
 

data/lib/stability_sdk/client.rb

@@ -72,10 +72,12 @@ module StabilitySDK
 
       prompt_param = []
       if prompt != ""
-        prompt_param << Gooseai::Prompt.new(text: prompt)
+        param = Gooseai::Prompt.new(text: prompt)
+        param.parameters = Gooseai::PromptParameters.new(weight: options[:prompt_weight]) if options.has_key?(:prompt_weight)
+        prompt_param << param
       end
       if options.has_key?(:init_image)
-        prompt_param << init_image_to_prompt(options[:init_image])
+        prompt_param << init_image_to_prompt(options[:init_image], options[:prompt_weight])
         step_parameter.scaled_step = 0
         step_parameter.sampler = Gooseai::SamplerParameters.new(
           cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale] : DEFAULT_CFG_SCALE,
@@ -168,17 +170,18 @@ module StabilitySDK
       end
     end
 
-    def init_image_to_prompt(path)
+    def init_image_to_prompt(path, weight)
       bin = IO.binread(path)
-      return Gooseai::Prompt.new(
+      prompt = Gooseai::Prompt.new(
         artifact: Gooseai::Artifact.new(
           type: Gooseai::ArtifactType::ARTIFACT_IMAGE,
           binary: bin,
-        ),
-        parameters: Gooseai::PromptParameters.new(
-          init: true
-        ),
+        )
       )
+      unless weight.nil?
+        prompt.parameters = Gooseai::PromptParameters.new(weight: weight)
+      end
+      return prompt
     end
 
     def mask_image_to_prompt(path)

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.3.4"
+  VERSION = "0.3.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.3.4
+  version: 0.3.5
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-11-07 00:00:00.000000000 Z
+date: 2023-11-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc