stability_sdk 0.3.3 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1bb93f79d9c09847cc8ee382f92cdf019137dbaa8b4c762bd1e24de392551a0f
-  data.tar.gz: a06818c77b6d490dec40047cf08b6b17e5ca72e09d8319499781dab3ef77a820
+  metadata.gz: d9c623b34b265936b3560c9bf2c8bc2f20f1882df4656c8050777cd2bb60b027
+  data.tar.gz: d81a22156c80f6f62cb4bb0a3bcd00094dae492ccca4a1e0d0a9e6a62d2e5870
 SHA512:
-  metadata.gz: 7ce00a0dcd937cdf0dbeb83b54d1437b5d26912b7139a99d125171f1364d3a03a377bef54698a9371b7b5dd44d6ad432bccfd01de9b02c3b7def1b01fec726f5
-  data.tar.gz: 84aa5aa534045145c9a4d6160b42e5cc2cdc5b41559e0c7122739f4720a00b36d757da57bab9cbef1b7936250e1a427b91014fa05b316ea45a33ade836c70261
+  metadata.gz: dcba5669c8bc711975c7f2c604c74e79396bf55b5594e1ce1ce3aff7627cf05eb356a3fb672e932670529b55aeaaecb97f02b595dbe274e3f02e27f4b16977ab
+  data.tar.gz: e0ee6ea802ba9ef6b664e3b98238d107c0dc1373e565a6e0a77ddac78c9e7466fb4664faa4beee737c3bbbd67ba53bb5d9ba9aeb72ad673b6a6e366aa41cc12d

data/README.md

@@ -56,7 +56,7 @@ Options:
     -p, --prefix=VAL                 output prefixes for artifacts. default `generation`
         --no-store                   do not write out artifacts
     -n, --num_samples=VAL            number of samples to generate. default 1
-    -e, --engine=VAL                 engine to use for inference. default `stable-diffusion-v1-5`
+    -e, --engine=VAL                 engine to use for inference. default `stable-diffusion-xl-1024-v1-0`
     -i, --init_image=VAL             path to init image
     -m, --mask_image=VAL             path to mask image
         --start_schedule=VAL         start schedule for init image (must be greater than 0, 1 is full strength text prompt, no trace of image). default 1.0
@@ -66,6 +66,8 @@ Options:
         --guidance_strength=VAL      Strength of the guidance. We recommend values in range [0.0,1.0]. A good default is 0.25. default nil
         --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

@@ -8,7 +8,7 @@ module StabilitySDK
     DEFAULT_IMAGE_WIDTH = 512
     DEFAULT_IMAGE_HEIGHT = 512
     DEFAULT_SAMPLE_SIZE = 1
-    DEFAULT_ENGINE_ID = "stable-diffusion-v1-5"
+    DEFAULT_ENGINE_ID = "stable-diffusion-xl-1024-v1-0"
     DEFAULT_CFG_SCALE = 7.0
     DEFAULT_START_SCHEDULE = 1.0
     DEFAULT_END_SCHEDULE = 0.01
@@ -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.3"
+  VERSION = "0.3.5"
 end

data/stability_sdk.gemspec

@@ -15,6 +15,7 @@ Gem::Specification.new do |spec|
   spec.metadata["homepage_uri"] = spec.homepage
   spec.metadata["source_code_uri"] = "https://github.com/cou929/stability-sdk-ruby"
   spec.metadata["changelog_uri"] = "https://github.com/cou929/stability-sdk-ruby"
+  spec.metadata["rubygems_mfa_required"] = "true"
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.3.3
+  version: 0.3.5
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-02-10 00:00:00.000000000 Z
+date: 2023-11-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -125,6 +125,7 @@ metadata:
   homepage_uri: https://github.com/cou929/stability-sdk-ruby
   source_code_uri: https://github.com/cou929/stability-sdk-ruby
   changelog_uri: https://github.com/cou929/stability-sdk-ruby
+  rubygems_mfa_required: 'true'
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -140,7 +141,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.1.6
 signing_key: 
 specification_version: 4
 summary: Ruby client for interacting with stability.ai APIs (e.g. stable diffusion