stability_sdk 0.2.10 → 0.2.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b45f00aa32974424860512f59fb3c1745b6c3b8b08f359e95ccd603f637642d
-  data.tar.gz: a67397bdc2b10b0bc49b7a508e3283c20fa4d7ea4ae4b4598b086b0ecc43b323
+  metadata.gz: d5aac33cb8d504aab26b515dc2fa6bd9d76c70829a0aa706eabc6be39043b143
+  data.tar.gz: ca6a3cd1cd24763a9f9427132a4e8fde4c688d9bc9d3409ef921139fa62fa232
 SHA512:
-  metadata.gz: f2fe617756b1d720e1d3a5c63807a7dbde52b70df2eb7a3c11a2bde803245f0e00c25b29d0d1893359de2243d333cf82ce274e23e03f220dadcad1d6f661261c
-  data.tar.gz: e50e38e2260d9e376581c03555468273c4378dea33cefdce1fb3e00e0d32db8aa592a63ba0e87f9685d0c89cdbd95c75af5093fc3e724711d862727d346028ba
+  metadata.gz: 95b76f6caf8e5a32af792cc0d284f81b76c7e38c682b24a5fc91204b3373096a47a98ab1ee6f0e32cbbd2039ad6b57c920c262661cda221c54eac529bd54e51a
+  data.tar.gz: f196c8b879b36e69c84d40c2528fd9fbf7f07e1ae6eec9c4a996a66a8a73841ea50d73d8ca941b7813ca9d5fbada94140143619bdb841de4ce9f71316af1303e

data/README.md

@@ -61,9 +61,25 @@ Options:
     -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
         --end_schedule=VAL           end schedule for init image. default 0.01
+        --guidance_preset=VAL        Guidance preset to use. See generation.GuidancePreset for supported values. default `GUIDANCE_PRESET_NONE`
+        --guidance_cuts=VAL          Number of cuts to use for guidance. default 0
+        --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
     -v, --verbose
 ```
 
+#### CLIP guidance
+
+By specifying `--guidance_preset` and other `--guidance_*` options, you can use CLIP guidance, which is enabled by default on DreamStudio's web interface.
+See also [this notebook](https://github.com/Stability-AI/stability-sdk/blob/c04381f960008f37c7392467cfaabfdf8f763e6a/nbs/demo_colab.ipynb) for more details.
+
+```sh
+# example
+# [note] --steps=35 is recommended. which is also the default value of the web interface
+STABILITY_SDK_API_KEY=xxx stability-client --guidance_preset=GUIDANCE_PRESET_FAST_BLUE --guidance_strength=0.25 --steps=35 --sampler=k_dpm_2_ancestral "A dream of a distant galaxy, by Caspar David Friedrich, matte painting trending on artstation HQ"
+```
+
 ### SDK usage
 
 This sample code saves a generated image as `result.png`.
@@ -91,12 +107,57 @@ client.generate(prompt, options) do |answer|
 end
 ```
 
+### [Unstable] Dashboard API
+
+**This feature is in a very early stage of development.**
+
+Dashboard API is a way to interact with DreamStudio Web UIs, such as getting user info, payment info, etc.
+
+Currently, there is no canonical way to get the API key for the Dashboard API. You can retrieve the key by logging in to the [DreamStudio Web page](https://beta.dreamstudio.ai/dream) and inspect request the authorization header with Chrome Developer Tool. Please be aware that how to refresh the key or an expiration period is unknown. See also https://github.com/Stability-AI/stability-sdk/issues/23
+
+```sh
+# get user info
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_me
+
+# get organization info
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization
+
+# i.e, get remaining balance
+STABILITY_SDK_DASHBOARD_API_KEY=YOUR_API_KEY stability-dashboard-client-unstable get_organization | jq .paymentInfo.balance
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
 To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
 
+### update submodule to try a new feature on Stability-AI/api-interfaces
+
+- update submodule
+
+```sh
+git submodule update --init  # if you haven't fetched the content of the submodule yet
+cd api-interfaces
+
+# checkout some branch/commit you need
+git fetch
+git reset --hard origin/some_branch
+
+cd ..
+```
+
+- build
+
+```sh
+bundle exec rake protoc
+
+git diff
+# now you may be able to confirm that the diff is created in lib/generation_pb.rb
+```
+
+- modify the `lib/stability_sdk/client.rb` to try some new features
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/cou929/stability-sdk-ruby.

data/Rakefile

@@ -13,3 +13,8 @@ desc "Parse proto file and generate output"
 task :protoc do
   sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/generation.proto"
 end
+
+desc "Compile a dashboard proto file"
+task :protoc_dashboard do
+  sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/dashboard.proto"
+end

data/exe/stability-client

@@ -18,20 +18,25 @@ opt.banner = "Usage: stability-client [options] YOUR_PROMPT_TEXT"
 opt.separator ""
 opt.separator "Options:"
 opt.on("--api_key=VAL", "api key of DreamStudio account. You can also specify by a STABILITY_SDK_API_KEY environment variable") {|v| options[:api_key] = v }
-opt.on("-H", "--height=VAL", "height of image in pixel. default 512") {|v| options[:height] = v }
-opt.on("-W", "--width=VAL", "width of image in pixel. default 512") {|v| options[:width] = v }
-opt.on("-C", "--cfg_scale=VAL", "CFG scale factor. default 7.0") {|v| options[:cfg_scale] = v }
-opt.on("-A", "--sampler=VAL", "ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_lms. default k_lms") {|v| options[:sampler] = v }
-opt.on("-s", "--steps=VAL", "number of steps. default 50") {|v| options[:steps] = v }
-opt.on("-S", "--seed=VAL", "random seed to use in integer") {|v| options[:seed] = v }
-opt.on("-p", "--prefix=VAL", "output prefixes for artifacts. default `generation`") {|v| options[:prefix] = v }
+opt.on("-H", "--height=VAL", Integer, "height of image in pixel. default 512") {|v| options[:height] = v }
+opt.on("-W", "--width=VAL", Integer, "width of image in pixel. default 512") {|v| options[:width] = v }
+opt.on("-C", "--cfg_scale=VAL", Float, "CFG scale factor. default 7.0") {|v| options[:cfg_scale] = v }
+opt.on("-A", "--sampler=VAL", String, "ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_lms. default k_lms") {|v| options[:sampler] = v }
+opt.on("-s", "--steps=VAL", Integer, "number of steps. default 50") {|v| options[:steps] = v }
+opt.on("-S", "--seed=VAL", Integer, "random seed to use in integer") {|v| options[:seed] = v }
+opt.on("-p", "--prefix=VAL", String, "output prefixes for artifacts. default `generation`") {|v| options[:prefix] = v }
 opt.on("--no-store", "do not write out artifacts") {|v| options[:no_store] = v }
-opt.on("-n", "--num_samples=VAL", "number of samples to generate") {|v| options[:num_samples] = v }
-opt.on("-e", "--engine=VAL", "engine to use for inference. default `stable-diffusion-v1`") {|v| options[:engine_id] = v }
-opt.on("-i", "--init_image=VAL", "path to init image") {|v| options[:init_image] = v }
-opt.on("-m", "--mask_image=VAL", "path to mask image") {|v| options[:mask_image] = v }
-opt.on("--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") {|v| options[:start_schedule] = v }
-opt.on("--end_schedule=VAL", "end schedule for init image. default 0.01") {|v| options[:end_schedule] = v }
+opt.on("-n", "--num_samples=VAL", Integer, "number of samples to generate. default 1") {|v| options[:num_samples] = v }
+opt.on("-e", "--engine=VAL", String, "engine to use for inference. default `stable-diffusion-v1-5`") {|v| options[:engine_id] = v }
+opt.on("-i", "--init_image=VAL", String, "path to init image") {|v| options[:init_image] = v }
+opt.on("-m", "--mask_image=VAL", String, "path to mask image") {|v| options[:mask_image] = v }
+opt.on("--start_schedule=VAL", Float, "start schedule for init image (must be greater than 0, 1 is full strength text prompt, no trace of image). default 1.0") {|v| options[:start_schedule] = v }
+opt.on("--end_schedule=VAL", Float, "end schedule for init image. default 0.01") {|v| options[:end_schedule] = v }
+opt.on("--guidance_preset=VAL", String,"Guidance preset to use. See generation.GuidancePreset for supported values. default `GUIDANCE_PRESET_NONE`") {|v| options[:guidance_preset] = v }
+opt.on("--guidance_cuts=VAL", Integer, "Number of cuts to use for guidance. default 0") {|v| options[:guidance_cuts] = v }
+opt.on("--guidance_strength=VAL", Float, "Strength of the guidance. We recommend values in range [0.0,1.0]. A good default is 0.25. default nil") {|v| options[:guidance_strength] = v }
+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("-v", "--verbose") { logger.level = Logger::DEBUG }
 opt.parse!(ARGV)
 
@@ -48,7 +53,7 @@ if options.has_key?(:init_image)
   raise StabilitySDK::InvalidParameter, "width and height of init_image must be a multiple of 64" if size[0] % 64 != 0 || size[1] % 64 != 0
 end
 
-client = StabilitySDK::Client.new(api_key: options[:api_key])
+client = StabilitySDK::Client.new(api_key: options[:api_key], logger: logger)
 
 client.generate(prompt, options) do |answer|
   StabilitySDK::CLI.save_answer(answer, options, logger)

data/exe/stability-dashboard-client-unstable

@@ -0,0 +1,40 @@
+#!/usr/bin/env ruby
+
+require "thor"
+require "logger"
+require "stability_sdk"
+
+logger = Logger.new(STDOUT)
+logger.level = Logger::WARN
+
+Version = StabilitySDK::VERSION
+
+class CLI < Thor
+  class_option :api_key
+
+  desc "get_me", "get user info"
+  def get_me
+    client = self.client(options)
+    res = client.get_me
+    puts res.to_json
+  end
+
+  desc "get_organization", "get organization info"
+  def get_organization
+    client = self.client(options)
+    res = client.get_organization
+    puts res.to_json
+  end
+
+  private
+
+  def client(options)
+    api_key = options[:api_key]
+    api_key = ENV["STABILITY_SDK_DASHBOARD_API_KEY"] if ENV["STABILITY_SDK_DASHBOARD_API_KEY"]
+    raise StabilitySDK::InsufficientParameter, "api key is required" if api_key.nil?
+
+    return StabilitySDK::Unstable::DashboardClient.new(api_key: api_key)
+  end
+end
+
+CLI.start(ARGV)

data/lib/dashboard_pb.rb

@@ -0,0 +1,182 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: dashboard.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("dashboard.proto", :syntax => :proto3) do
+    add_message "gooseai.OrganizationMember" do
+      optional :organization, :message, 1, "gooseai.Organization"
+      proto3_optional :user, :message, 2, "gooseai.User"
+      optional :role, :enum, 3, "gooseai.OrganizationRole"
+      optional :is_default, :bool, 4
+    end
+    add_message "gooseai.OrganizationGrant" do
+      optional :amount_granted, :double, 1
+      optional :amount_used, :double, 2
+      optional :expires_at, :uint64, 3
+      optional :granted_at, :uint64, 4
+    end
+    add_message "gooseai.OrganizationPaymentInfo" do
+      optional :balance, :double, 1
+      repeated :grants, :message, 2, "gooseai.OrganizationGrant"
+    end
+    add_message "gooseai.OrganizationAutoCharge" do
+      optional :enabled, :bool, 1
+      optional :id, :string, 2
+      optional :created_at, :uint64, 3
+    end
+    add_message "gooseai.Organization" do
+      optional :id, :string, 1
+      optional :name, :string, 2
+      optional :description, :string, 3
+      repeated :members, :message, 4, "gooseai.OrganizationMember"
+      proto3_optional :payment_info, :message, 5, "gooseai.OrganizationPaymentInfo"
+      proto3_optional :stripe_customer_id, :string, 6
+      proto3_optional :auto_charge, :message, 7, "gooseai.OrganizationAutoCharge"
+    end
+    add_message "gooseai.APIKey" do
+      optional :key, :string, 1
+      optional :is_secret, :bool, 2
+      optional :created_at, :uint64, 3
+    end
+    add_message "gooseai.User" do
+      optional :id, :string, 1
+      proto3_optional :auth_id, :string, 2
+      optional :profile_picture, :string, 3
+      optional :email, :string, 4
+      repeated :organizations, :message, 5, "gooseai.OrganizationMember"
+      repeated :api_keys, :message, 7, "gooseai.APIKey"
+      optional :created_at, :uint64, 8
+      proto3_optional :email_verified, :bool, 9
+    end
+    add_message "gooseai.CostData" do
+      optional :amount_tokens, :uint32, 1
+      optional :amount_credits, :double, 2
+    end
+    add_message "gooseai.UsageMetric" do
+      optional :operation, :string, 1
+      optional :engine, :string, 2
+      optional :input_cost, :message, 3, "gooseai.CostData"
+      optional :output_cost, :message, 4, "gooseai.CostData"
+      proto3_optional :user, :string, 5
+      optional :aggregation_timestamp, :uint64, 6
+    end
+    add_message "gooseai.CostTotal" do
+      optional :amount_tokens, :uint32, 1
+      optional :amount_credits, :double, 2
+    end
+    add_message "gooseai.TotalMetricsData" do
+      optional :input_total, :message, 1, "gooseai.CostTotal"
+      optional :output_total, :message, 2, "gooseai.CostTotal"
+    end
+    add_message "gooseai.Metrics" do
+      repeated :metrics, :message, 1, "gooseai.UsageMetric"
+      optional :total, :message, 2, "gooseai.TotalMetricsData"
+    end
+    add_message "gooseai.EmptyRequest" do
+    end
+    add_message "gooseai.GetOrganizationRequest" do
+      optional :id, :string, 1
+    end
+    add_message "gooseai.GetMetricsRequest" do
+      optional :organization_id, :string, 1
+      proto3_optional :user_id, :string, 2
+      optional :range_from, :uint64, 3
+      optional :range_to, :uint64, 4
+      optional :include_per_request_metrics, :bool, 5
+    end
+    add_message "gooseai.APIKeyRequest" do
+      optional :is_secret, :bool, 1
+    end
+    add_message "gooseai.APIKeyFindRequest" do
+      optional :id, :string, 1
+    end
+    add_message "gooseai.UpdateDefaultOrganizationRequest" do
+      optional :organization_id, :string, 1
+    end
+    add_message "gooseai.ClientSettings" do
+      optional :settings, :bytes, 1
+    end
+    add_message "gooseai.CreateAutoChargeIntentRequest" do
+      optional :organization_id, :string, 1
+      optional :monthly_maximum, :uint64, 2
+      optional :minimum_value, :uint64, 3
+      optional :amount_credits, :uint64, 4
+    end
+    add_message "gooseai.CreateChargeRequest" do
+      optional :amount, :uint64, 1
+      optional :organization_id, :string, 2
+    end
+    add_message "gooseai.GetChargesRequest" do
+      optional :organization_id, :string, 1
+      optional :range_from, :uint64, 2
+      optional :range_to, :uint64, 3
+    end
+    add_message "gooseai.Charge" do
+      optional :id, :string, 1
+      optional :paid, :bool, 2
+      optional :receipt_link, :string, 3
+      optional :payment_link, :string, 4
+      optional :created_at, :uint64, 5
+      optional :amount_credits, :uint64, 6
+    end
+    add_message "gooseai.Charges" do
+      repeated :charges, :message, 1, "gooseai.Charge"
+    end
+    add_message "gooseai.GetAutoChargeRequest" do
+      optional :organization_id, :string, 1
+    end
+    add_message "gooseai.AutoChargeIntent" do
+      optional :id, :string, 1
+      optional :payment_link, :string, 2
+      optional :created_at, :uint64, 3
+      optional :monthly_maximum, :uint64, 4
+      optional :minimum_value, :uint64, 5
+      optional :amount_credits, :uint64, 6
+    end
+    add_message "gooseai.UpdateUserInfoRequest" do
+      proto3_optional :email, :string, 1
+    end
+    add_message "gooseai.UserPasswordChangeTicket" do
+      optional :ticket, :string, 1
+    end
+    add_enum "gooseai.OrganizationRole" do
+      value :MEMBER, 0
+      value :ACCOUNTANT, 1
+      value :OWNER, 2
+    end
+  end
+end
+
+module Gooseai
+  OrganizationMember = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationMember").msgclass
+  OrganizationGrant = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationGrant").msgclass
+  OrganizationPaymentInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationPaymentInfo").msgclass
+  OrganizationAutoCharge = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationAutoCharge").msgclass
+  Organization = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Organization").msgclass
+  APIKey = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKey").msgclass
+  User = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.User").msgclass
+  CostData = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CostData").msgclass
+  UsageMetric = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UsageMetric").msgclass
+  CostTotal = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CostTotal").msgclass
+  TotalMetricsData = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.TotalMetricsData").msgclass
+  Metrics = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Metrics").msgclass
+  EmptyRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.EmptyRequest").msgclass
+  GetOrganizationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetOrganizationRequest").msgclass
+  GetMetricsRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetMetricsRequest").msgclass
+  APIKeyRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKeyRequest").msgclass
+  APIKeyFindRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.APIKeyFindRequest").msgclass
+  UpdateDefaultOrganizationRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UpdateDefaultOrganizationRequest").msgclass
+  ClientSettings = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClientSettings").msgclass
+  CreateAutoChargeIntentRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CreateAutoChargeIntentRequest").msgclass
+  CreateChargeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CreateChargeRequest").msgclass
+  GetChargesRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetChargesRequest").msgclass
+  Charge = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Charge").msgclass
+  Charges = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Charges").msgclass
+  GetAutoChargeRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GetAutoChargeRequest").msgclass
+  AutoChargeIntent = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AutoChargeIntent").msgclass
+  UpdateUserInfoRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UpdateUserInfoRequest").msgclass
+  UserPasswordChangeTicket = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.UserPasswordChangeTicket").msgclass
+  OrganizationRole = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OrganizationRole").enummodule
+end

data/lib/dashboard_services_pb.rb

@@ -0,0 +1,41 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: dashboard.proto for package 'gooseai'
+
+require 'grpc'
+require 'dashboard_pb'
+
+module Gooseai
+  module DashboardService
+    class Service
+
+      include ::GRPC::GenericService
+
+      self.marshal_class_method = :encode
+      self.unmarshal_class_method = :decode
+      self.service_name = 'gooseai.DashboardService'
+
+      # Get info
+      rpc :GetMe, ::Gooseai::EmptyRequest, ::Gooseai::User
+      rpc :GetOrganization, ::Gooseai::GetOrganizationRequest, ::Gooseai::Organization
+      rpc :GetMetrics, ::Gooseai::GetMetricsRequest, ::Gooseai::Metrics
+      # API key management
+      rpc :CreateAPIKey, ::Gooseai::APIKeyRequest, ::Gooseai::APIKey
+      rpc :DeleteAPIKey, ::Gooseai::APIKeyFindRequest, ::Gooseai::APIKey
+      # User settings
+      rpc :UpdateDefaultOrganization, ::Gooseai::UpdateDefaultOrganizationRequest, ::Gooseai::User
+      rpc :GetClientSettings, ::Gooseai::EmptyRequest, ::Gooseai::ClientSettings
+      rpc :SetClientSettings, ::Gooseai::ClientSettings, ::Gooseai::ClientSettings
+      rpc :UpdateUserInfo, ::Gooseai::UpdateUserInfoRequest, ::Gooseai::User
+      rpc :CreatePasswordChangeTicket, ::Gooseai::EmptyRequest, ::Gooseai::UserPasswordChangeTicket
+      rpc :DeleteAccount, ::Gooseai::EmptyRequest, ::Gooseai::User
+      # Payment functions
+      rpc :CreateCharge, ::Gooseai::CreateChargeRequest, ::Gooseai::Charge
+      rpc :GetCharges, ::Gooseai::GetChargesRequest, ::Gooseai::Charges
+      rpc :CreateAutoChargeIntent, ::Gooseai::CreateAutoChargeIntentRequest, ::Gooseai::AutoChargeIntent
+      rpc :UpdateAutoChargeIntent, ::Gooseai::CreateAutoChargeIntentRequest, ::Gooseai::AutoChargeIntent
+      rpc :GetAutoChargeIntent, ::Gooseai::GetAutoChargeRequest, ::Gooseai::AutoChargeIntent
+    end
+
+    Stub = Service.rpc_stub_class
+  end
+end

data/lib/generation_pb.rb

@@ -21,6 +21,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       optional :index, :uint32, 8
       optional :finish_reason, :enum, 9, "gooseai.FinishReason"
       optional :seed, :uint32, 10
+      optional :uuid, :string, 12
+      optional :size, :uint64, 13
       oneof :data do
         optional :binary, :bytes, 5
         optional :text, :string, 6
@@ -40,20 +42,6 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :artifact, :message, 4, "gooseai.Artifact"
       end
     end
-    add_message "gooseai.AnswerMeta" do
-      proto3_optional :gpu_id, :string, 1
-      proto3_optional :cpu_id, :string, 2
-      proto3_optional :node_id, :string, 3
-      proto3_optional :engine_id, :string, 4
-    end
-    add_message "gooseai.Answer" do
-      optional :answer_id, :string, 1
-      optional :request_id, :string, 2
-      optional :received, :uint64, 3
-      optional :created, :uint64, 4
-      proto3_optional :meta, :message, 6, "gooseai.AnswerMeta"
-      repeated :artifacts, :message, 7, "gooseai.Artifact"
-    end
     add_message "gooseai.SamplerParameters" do
       proto3_optional :eta, :float, 1
       proto3_optional :sampling_steps, :uint64, 2
@@ -63,15 +51,48 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
     end
     add_message "gooseai.ConditionerParameters" do
       proto3_optional :vector_adjust_prior, :string, 1
+      proto3_optional :conditioner, :message, 2, "gooseai.Model"
     end
     add_message "gooseai.ScheduleParameters" do
       proto3_optional :start, :float, 1
       proto3_optional :end, :float, 2
+      proto3_optional :value, :float, 3
     end
     add_message "gooseai.StepParameter" do
       optional :scaled_step, :float, 1
       proto3_optional :sampler, :message, 2, "gooseai.SamplerParameters"
       proto3_optional :schedule, :message, 3, "gooseai.ScheduleParameters"
+      proto3_optional :guidance, :message, 4, "gooseai.GuidanceParameters"
+    end
+    add_message "gooseai.Model" do
+      optional :architecture, :enum, 1, "gooseai.ModelArchitecture"
+      optional :publisher, :string, 2
+      optional :dataset, :string, 3
+      optional :version, :float, 4
+      optional :semantic_version, :string, 5
+      optional :alias, :string, 6
+    end
+    add_message "gooseai.CutoutParameters" do
+      repeated :cutouts, :message, 1, "gooseai.CutoutParameters"
+      proto3_optional :count, :uint32, 2
+      proto3_optional :gray, :float, 3
+      proto3_optional :blur, :float, 4
+      proto3_optional :size_power, :float, 5
+    end
+    add_message "gooseai.GuidanceScheduleParameters" do
+      optional :duration, :float, 1
+      optional :value, :float, 2
+    end
+    add_message "gooseai.GuidanceInstanceParameters" do
+      repeated :models, :message, 2, "gooseai.Model"
+      proto3_optional :guidance_strength, :float, 3
+      repeated :schedule, :message, 4, "gooseai.GuidanceScheduleParameters"
+      proto3_optional :cutouts, :message, 5, "gooseai.CutoutParameters"
+      proto3_optional :prompt, :message, 6, "gooseai.Prompt"
+    end
+    add_message "gooseai.GuidanceParameters" do
+      optional :guidance_preset, :enum, 1, "gooseai.GuidancePreset"
+      repeated :instances, :message, 2, "gooseai.GuidanceInstanceParameters"
     end
     add_message "gooseai.TransformType" do
       oneof :type do
@@ -104,6 +125,25 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       repeated :exceeds, :message, 2, "gooseai.ClassifierCategory"
       proto3_optional :realized_action, :enum, 3, "gooseai.Action"
     end
+    add_message "gooseai.AssetParameters" do
+      optional :action, :enum, 1, "gooseai.AssetAction"
+      optional :project_id, :string, 2
+      optional :use, :enum, 3, "gooseai.AssetUse"
+    end
+    add_message "gooseai.AnswerMeta" do
+      proto3_optional :gpu_id, :string, 1
+      proto3_optional :cpu_id, :string, 2
+      proto3_optional :node_id, :string, 3
+      proto3_optional :engine_id, :string, 4
+    end
+    add_message "gooseai.Answer" do
+      optional :answer_id, :string, 1
+      optional :request_id, :string, 2
+      optional :received, :uint64, 3
+      optional :created, :uint64, 4
+      proto3_optional :meta, :message, 6, "gooseai.AnswerMeta"
+      repeated :artifacts, :message, 7, "gooseai.Artifact"
+    end
     add_message "gooseai.Request" do
       optional :engine_id, :string, 1
       optional :request_id, :string, 2
@@ -113,6 +153,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       oneof :params do
         optional :image, :message, 5, "gooseai.ImageParameters"
         optional :classifier, :message, 7, "gooseai.ClassifierParameters"
+        optional :asset, :message, 8, "gooseai.AssetParameters"
       end
     end
     add_message "gooseai.OnStatus" do
@@ -161,6 +202,21 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :UPSCALER_GFPGAN, 1
       value :UPSCALER_ESRGAN, 2
     end
+    add_enum "gooseai.GuidancePreset" do
+      value :GUIDANCE_PRESET_NONE, 0
+      value :GUIDANCE_PRESET_SIMPLE, 1
+      value :GUIDANCE_PRESET_FAST_BLUE, 2
+      value :GUIDANCE_PRESET_FAST_GREEN, 3
+      value :GUIDANCE_PRESET_SLOW, 4
+      value :GUIDANCE_PRESET_SLOWER, 5
+      value :GUIDANCE_PRESET_SLOWEST, 6
+    end
+    add_enum "gooseai.ModelArchitecture" do
+      value :MODEL_ARCHITECTURE_NONE, 0
+      value :MODEL_ARCHITECTURE_CLIP_VIT, 1
+      value :MODEL_ARCHITECTURE_CLIP_RESNET, 2
+      value :MODEL_ARCHITECTURE_LDM, 3
+    end
     add_enum "gooseai.Action" do
       value :ACTION_PASSTHROUGH, 0
       value :ACTION_REGENERATE_DUPLICATE, 1
@@ -173,6 +229,18 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :CLSFR_MODE_ZEROSHOT, 0
       value :CLSFR_MODE_MULTICLASS, 1
     end
+    add_enum "gooseai.AssetAction" do
+      value :ASSET_PUT, 0
+      value :ASSET_GET, 1
+      value :ASSET_DELETE, 2
+    end
+    add_enum "gooseai.AssetUse" do
+      value :ASSET_USE_UNDEFINED, 0
+      value :ASSET_USE_INPUT, 1
+      value :ASSET_USE_OUTPUT, 2
+      value :ASSET_USE_INTERMEDIATE, 3
+      value :ASSET_USE_PROJECT, 4
+    end
     add_enum "gooseai.StageAction" do
       value :STAGE_ACTION_PASS, 0
       value :STAGE_ACTION_DISCARD, 1
@@ -187,17 +255,23 @@ module Gooseai
   Artifact = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Artifact").msgclass
   PromptParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.PromptParameters").msgclass
   Prompt = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Prompt").msgclass
-  AnswerMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AnswerMeta").msgclass
-  Answer = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Answer").msgclass
   SamplerParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.SamplerParameters").msgclass
   ConditionerParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ConditionerParameters").msgclass
   ScheduleParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ScheduleParameters").msgclass
   StepParameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StepParameter").msgclass
+  Model = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Model").msgclass
+  CutoutParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.CutoutParameters").msgclass
+  GuidanceScheduleParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceScheduleParameters").msgclass
+  GuidanceInstanceParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceInstanceParameters").msgclass
+  GuidanceParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidanceParameters").msgclass
   TransformType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.TransformType").msgclass
   ImageParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ImageParameters").msgclass
   ClassifierConcept = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierConcept").msgclass
   ClassifierCategory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierCategory").msgclass
   ClassifierParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierParameters").msgclass
+  AssetParameters = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetParameters").msgclass
+  AnswerMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AnswerMeta").msgclass
+  Answer = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Answer").msgclass
   Request = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Request").msgclass
   OnStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.OnStatus").msgclass
   Stage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Stage").msgclass
@@ -206,7 +280,11 @@ module Gooseai
   ArtifactType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ArtifactType").enummodule
   DiffusionSampler = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.DiffusionSampler").enummodule
   Upscaler = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Upscaler").enummodule
+  GuidancePreset = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.GuidancePreset").enummodule
+  ModelArchitecture = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ModelArchitecture").enummodule
   Action = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Action").enummodule
   ClassifierMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierMode").enummodule
+  AssetAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetAction").enummodule
+  AssetUse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.AssetUse").enummodule
   StageAction = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StageAction").enummodule
 end

data/lib/generation_services_pb.rb

@@ -6,6 +6,9 @@ require 'generation_pb'
 
 module Gooseai
   module GenerationService
+    #
+    # gRPC services
+    #
     class Service
 
       include ::GRPC::GenericService

data/lib/stability_sdk/cli.rb

@@ -23,10 +23,15 @@ module StabilitySDK
           logger.warn "not implemented for ArtifactType #{artifact.type}"
         end
 
+        if artifact.finish_reason == :FILTER
+          logger.debug "the generated image is filtered"
+        end
+
         next if filename == "" || contents == ""
 
         File.open(filename, "wb") do |f|
           f.write(contents)
+          logger.debug "wrote #{artifact.type} to #{filename}"
         end
       end
     end

data/lib/stability_sdk/client.rb

@@ -1,5 +1,6 @@
 require "grpc"
 require "generation_services_pb"
+require "logger"
 
 module StabilitySDK
   class Client
@@ -36,29 +37,37 @@ module StabilitySDK
         stub_params[kw] = options[kw] if options.has_key?(kw)
       end
 
+      if options.has_key?(:logger)
+        @logger = options[:logger]
+      else
+        logger = Logger.new(STDOUT)
+        logger.level = Logger::WARN
+        @logger = logger
+      end
+
       @stub = Gooseai::GenerationService::Stub.new(host, creds, **stub_params)
     end
 
     def generate(prompt, options, &block)
-      width = options.has_key?(:width) ? options[:width].to_i : DEFAULT_IMAGE_WIDTH
-      height = options.has_key?(:height) ? options[:height].to_i : DEFAULT_IMAGE_HEIGHT
+      width = options.has_key?(:width) ? options[:width] : DEFAULT_IMAGE_WIDTH
+      height = options.has_key?(:height) ? options[:height] : DEFAULT_IMAGE_HEIGHT
 
       if width % 64 != 0 || height % 64 != 0
         raise InvalidParameter, "width and height must be a multiple of 64"
       end
 
-      samples = options.has_key?(:num_samples) ? options[:num_samples].to_i : DEFAULT_SAMPLE_SIZE
-      steps = options.has_key?(:steps) ? options[:steps].to_i : DEFAULT_STEPS
-      seed = options.has_key?(:seed) ? [options[:seed].to_i] : [rand(4294967295)]
+      samples = options.has_key?(:num_samples) ? options[:num_samples] : DEFAULT_SAMPLE_SIZE
+      steps = options.has_key?(:steps) ? options[:steps] : DEFAULT_STEPS
+      seed = options.has_key?(:seed) ? [options[:seed]] : [rand(4294967295)]
       transform = Gooseai::TransformType.new(
-        diffusion: options.has_key?(:sampler) ? SAMPLER_ALGORITHMS[options[:sampler]] : DEFAULT_SAMPLER_ALGORITHM,
+        diffusion: options.has_key?(:sampler) ? SAMPLER_ALGORITHMS[options[:sampler].to_sym] : DEFAULT_SAMPLER_ALGORITHM,
       )
-      parameters = [Gooseai::StepParameter.new(
+      step_parameter = Gooseai::StepParameter.new(
         scaled_step: 0,
         sampler: Gooseai::SamplerParameters.new(
-          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale].to_f : DEFAULT_CFG_SCALE,
+          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale] : DEFAULT_CFG_SCALE,
         ),
-      )]
+      )
 
       prompt_param = []
       if prompt != ""
@@ -66,21 +75,65 @@ module StabilitySDK
       end
       if options.has_key?(:init_image)
         prompt_param << init_image_to_prompt(options[:init_image])
-        parameters = [Gooseai::StepParameter.new(
-          scaled_step: 0,
-          sampler: Gooseai::SamplerParameters.new(
-            cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale].to_f : DEFAULT_CFG_SCALE,
-          ),
-          schedule: Gooseai::ScheduleParameters.new(
-            start: options.has_key?(:start_schedule) ? options[:start_schedule].to_f : DEFAULT_START_SCHEDULE,
-            end: options.has_key?(:end_schedule) ? options[:end_schedule].to_f : DEFAULT_END_SCHEDULE,
-          ),
-        )]
+        step_parameter.scaled_step = 0
+        step_parameter.sampler = Gooseai::SamplerParameters.new(
+          cfg_scale: options.has_key?(:cfg_scale) ? options[:cfg_scale] : DEFAULT_CFG_SCALE,
+        )
+        step_parameter.schedule = Gooseai::ScheduleParameters.new(
+          start: options.has_key?(:start_schedule) ? options[:start_schedule] : DEFAULT_START_SCHEDULE,
+          end: options.has_key?(:end_schedule) ? options[:end_schedule] : DEFAULT_END_SCHEDULE,
+        )
       end
       if options.has_key?(:mask_image)
         prompt_param << mask_image_to_prompt(options[:mask_image])
       end
 
+      # CLIP guidance
+      if options.has_key?(:guidance_preset) && options[:guidance_preset] != "GUIDANCE_PRESET_NONE"
+        step_parameter.sampler = nil
+
+        guidance_prompt =
+          if options.has_key?(:guidance_prompt)
+            Gooseai::Prompt.new(text: options[:guidance_prompt])
+          else
+            Gooseai::Prompt.new(text: prompt)
+          end
+
+        guidance_strength = nil
+        if options.has_key?(:guidance_strength) && options[:guidance_strength] != 0
+          guidance_strength = options[:guidance_strength]
+        end
+
+        models = nil
+        if options.has_key?(:guidance_models)
+          models = options[:guidance_models].map { |m| Gooseai::Model.new(alias: m) }
+        end
+
+        cutouts = nil
+        if options.has_key?(:guidance_cuts)
+          cutouts = Gooseai::CutoutParameters.new(count: options[:guidance_cuts])
+        end
+
+        step_parameter.guidance = Gooseai::GuidanceParameters.new(
+          guidance_preset: Gooseai::GuidancePreset.const_get(options[:guidance_preset].to_sym),
+          instances: [
+            Gooseai::GuidanceInstanceParameters.new(
+              guidance_strength: guidance_strength,
+              models: models,
+              cutouts: cutouts,
+              prompt: guidance_prompt,
+            ),
+          ],
+        )
+
+        if transform.diffusion != Gooseai::DiffusionSampler::SAMPLER_K_DPM_2_ANCESTRAL && transform.diffusion != Gooseai::DiffusionSampler::SAMPLER_K_EULER_ANCESTRAL
+          transform = Gooseai::TransformType.new(
+            diffusion: Gooseai::DiffusionSampler::SAMPLER_K_DPM_2_ANCESTRAL
+          )
+          @logger.warn "CLIP guidance is only supported with ancestral samplers. So override it with SAMPLER_K_DPM_2_ANCESTRAL."
+        end
+      end
+
       image_param = Gooseai::ImageParameters.new(
         width: width,
         height: height,
@@ -88,7 +141,7 @@ module StabilitySDK
         steps: steps,
         seed: seed,
         transform: transform,
-        parameters: parameters,
+        parameters: [step_parameter],
       )
 
       req = Gooseai::Request.new(
@@ -97,8 +150,20 @@ module StabilitySDK
         image: image_param
       )
 
+      @logger.debug "sending request."
+      start = Time.now
       @stub.generate(req).each do |answer|
+        duration = Time.now - start
+        if answer.artifacts.size > 0
+          artifact_types = answer.artifacts.map { |a| a.type }
+          @logger.debug "got #{answer.answer_id} with #{artifact_types} in #{duration.round(2)}s"
+        else
+          @logger.debug "got keepalive #{answer.answer_id} in #{duration.round(2)}s"
+        end
+
         block.call(answer)
+
+        start = Time.now
       end
     end
 

data/lib/stability_sdk/unstable/dashboard_client.rb

@@ -0,0 +1,31 @@
+require "grpc"
+require "dashboard_services_pb"
+
+module StabilitySDK::Unstable
+  class DashboardClient
+    DEFAULT_API_HOST = "grpc.stability.ai:443"
+
+    def initialize(options={})
+      host = options[:api_host] || DEFAULT_API_HOST
+      channel_creds = options.has_key?(:ca_cert) ? GRPC::Core::ChannelCredentials.new(options[:ca_cert]) : GRPC::Core::ChannelCredentials.new
+      call_creds = GRPC::Core::CallCredentials.new(proc { { "authorization" => "Bearer #{options[:api_key]}" } })
+      creds = channel_creds.compose(call_creds)
+
+      stub_params = {}
+      [:channel_override, :timeout, :propagate_mask, :channel_args, :interceptors].each do |kw|
+        stub_params[kw] = options[kw] if options.has_key?(kw)
+      end
+
+      @stub = Gooseai::DashboardService::Stub.new(host, creds, **stub_params)
+    end
+
+    def get_me
+      @stub.get_me(Gooseai::EmptyRequest.new)
+    end
+
+    def get_organization
+      me = self.get_me
+      @stub.get_organization(Gooseai::GetChargesRequest.new(organization_id: me.organizations[0].organization.id))
+    end
+  end
+end

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.2.10"
+  VERSION = "0.2.12"
 end

data/lib/stability_sdk.rb

@@ -1,6 +1,7 @@
 require "stability_sdk/version"
 require "stability_sdk/client"
 require "stability_sdk/cli"
+require "stability_sdk/unstable/dashboard_client"
 
 module StabilitySDK
   class Error < StandardError; end

data/stability_sdk.gemspec

@@ -28,5 +28,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency "grpc", ">= 1.41.1"
   spec.add_dependency "mime-types", ">= 3.0.0"
   spec.add_dependency "fastimage", "~> 2.2", ">= 2.2.6"
+  spec.add_dependency "thor", ">= 1.2.1"
   spec.add_development_dependency "grpc-tools"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.2.10
+  version: 0.2.12
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-09 00:00:00.000000000 Z
+date: 2022-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -58,6 +58,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 2.2.6
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
 - !ruby/object:Gem::Dependency
   name: grpc-tools
   requirement: !ruby/object:Gem::Requirement
@@ -78,6 +92,7 @@ email:
 - cou929@gmail.com
 executables:
 - stability-client
+- stability-dashboard-client-unstable
 extensions: []
 extra_rdoc_files: []
 files:
@@ -91,11 +106,15 @@ files:
 - bin/console
 - bin/setup
 - exe/stability-client
+- exe/stability-dashboard-client-unstable
+- lib/dashboard_pb.rb
+- lib/dashboard_services_pb.rb
 - lib/generation_pb.rb
 - lib/generation_services_pb.rb
 - lib/stability_sdk.rb
 - lib/stability_sdk/cli.rb
 - lib/stability_sdk/client.rb
+- lib/stability_sdk/unstable/dashboard_client.rb
 - lib/stability_sdk/version.rb
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby
@@ -120,7 +139,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