stability_sdk 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: dac029de10ccef05e11d7fee6a581f14fec50252e1cf7790ce2e849e5061489c
+  data.tar.gz: 66f8803ff804dae6971cdf3b8473225012255e87d67f00484583d65ecd8afb6a
+SHA512:
+  metadata.gz: ad988347184eafe784ab654b0022f1742cf56e51c9e2d4d88b3507f97162c6aaa7a165ad8ee5a6d3c34c1a9427fc860a54679192235c154bfea3008765dd8f89
+  data.tar.gz: d4522e6b2644f0ef309dbfb862a0667ed80c77e93c30bc575764997aa511d73789b3f270641d44c83cc2fdc017b7a0d5562e7de79ed7bebf86ff894019ad2642

data/.gitignore

@@ -0,0 +1,8 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/.travis.yml

@@ -0,0 +1,6 @@
+---
+language: ruby
+cache: bundler
+rvm:
+  - 2.7.2
+before_install: gem install bundler -v 2.1.4

data/Gemfile

@@ -0,0 +1,7 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in stability_sdk.gemspec
+gemspec
+
+gem "rake", "~> 12.0"
+gem "minitest", "~> 5.0"

data/Gemfile.lock

@@ -0,0 +1,34 @@
+PATH
+  remote: .
+  specs:
+    stability_sdk (0.1.0)
+      grpc
+      mime-types
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    google-protobuf (3.21.5)
+    googleapis-common-protos-types (1.4.0)
+      google-protobuf (~> 3.14)
+    grpc (1.48.0)
+      google-protobuf (~> 3.19)
+      googleapis-common-protos-types (~> 1.0)
+    grpc-tools (1.48.0)
+    mime-types (3.4.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2022.0105)
+    minitest (5.16.3)
+    rake (12.3.3)
+
+PLATFORMS
+  arm64-darwin-20
+
+DEPENDENCIES
+  grpc-tools
+  minitest (~> 5.0)
+  rake (~> 12.0)
+  stability_sdk!
+
+BUNDLED WITH
+   2.2.2

data/README.md

@@ -0,0 +1,89 @@
+# StabilitySDK - Ruby client for stability.ai APIs, such as Stable Diffusion
+
+A ruby client for [stability.ai](https://stability.ai/) APIs, e.g., [Stable Diffusion](https://stability.ai/blog/stable-diffusion-public-release). Referring to https://github.com/Stability-AI/stability-sdk.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'stability_sdk'
+```
+
+And then execute:
+
+    $ bundle install
+
+Or install it yourself as:
+
+    $ gem install stability_sdk
+
+## Usage
+
+First you need to create a [DreamStudio](https://beta.dreamstudio.ai/home)'s account and get an API Key of it.
+
+- Access [DreamStudio](https://beta.dreamstudio.ai/dream) and create an account if you have not had it
+- Go to the [membership page](https://beta.dreamstudio.ai/membership)
+- You can get the API Key in an `API Key` tab
+
+### Command line usage
+
+```sh
+STABILITY_SDK_API_KEY=YOUR_API_KEY stability-client 'A night in winter, oil-on-canvas landscape painting, by Vincent van Gogh'
+```
+
+This command saves an image like this:
+
+![3749380973_A_night_in_winter__oil_on_canvas_landscape_painting__by_Vincent_van_Gogh](https://user-images.githubusercontent.com/25668/188884116-0b03494b-0b34-49de-bbbc-89fbc2f6029d.png)
+
+
+```sh
+Usage: stability-client [options] YOUR_PROMPT_TEXT
+
+Options:
+        --api_key=VAL                api key of DreamStudio account. You can also specify by a STABILITY_SDK_API_KEY environment variable
+    -H, --height=VAL                 height of image in pixel. default 512
+    -W, --width=VAL                  width of image in pixel. default 512
+    -C, --cfg_scale=VAL              CFG scale factor. default 7.0
+    -A, --sampler=VAL                ddim, plms, k_euler, k_euler_ancestral, k_heun, k_dpm_2, k_dpm_2_ancestral, k_lms. default k_lms
+    -s, --steps=VAL                  number of steps. default 50
+    -S, --seed=VAL                   random seed to use in integer
+    -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
+    -e, --engine=VAL                 engine to use for inference. default `stable-diffusion-v1`
+    -v, --verbose
+```
+
+### SDK usage
+
+This sample code saves a generated image as `result.png`.
+
+```ruby
+require "stability_sdk"
+
+client = StabilitySDK::Client.new(api_key: "YOUR_API_KEY")
+
+prompt = "your prompot here"
+options = {}
+
+client.generate(prompt, options) do |answer|
+  answer.artifacts.each do |artifact|
+    if artifact.type == :ARTIFACT_IMAGE
+      File.open("result.png", "wb") do |f|
+        f.write(artifact.binary)
+      end
+    end
+  end
+end
+```
+
+## 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).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/cou929/stability-sdk-ruby.

data/Rakefile

@@ -0,0 +1,10 @@
+require "bundler/gem_tasks"
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << "test"
+  t.libs << "lib"
+  t.test_files = FileList["test/**/*_test.rb"]
+end
+
+task :default => :test

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "stability_sdk"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/exe/stability-client

@@ -0,0 +1,43 @@
+#!/usr/bin/env ruby
+
+require "optparse"
+require "mime/types"
+require "logger"
+require "stability_sdk"
+
+logger = Logger.new(STDOUT)
+logger.level = Logger::WARN
+
+opt = OptionParser.new
+Version = StabilitySDK::VERSION
+
+options = {}
+
+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("--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("-v", "--verbose") { logger.level = Logger::DEBUG }
+opt.parse!(ARGV)
+
+prompt = ARGV.join(" ")
+raise StabilitySDK::InsufficientParameter, "prompt is required" if prompt.nil? || prompt == ""
+
+options[:api_key] = ENV["STABILITY_SDK_API_KEY"] if ENV["STABILITY_SDK_API_KEY"]
+raise StabilitySDK::InsufficientParameter, "api key is required" if !options.has_key?(:api_key) || options[:api_key] == ""
+
+client = StabilitySDK::Client.new(api_key: options[:api_key])
+
+client.generate(prompt, options) do |answer|
+  StabilitySDK::CLI.save_answer(answer, options, logger)
+end

data/lib/generation_pb.rb

@@ -0,0 +1,182 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: generation.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("generation.proto", :syntax => :proto3) do
+    add_message "gooseai.Token" do
+      proto3_optional :text, :string, 1
+      optional :id, :uint32, 2
+    end
+    add_message "gooseai.Tokens" do
+      repeated :tokens, :message, 1, "gooseai.Token"
+      proto3_optional :tokenizer_id, :string, 2
+    end
+    add_message "gooseai.Artifact" do
+      optional :id, :uint64, 1
+      optional :type, :enum, 2, "gooseai.ArtifactType"
+      optional :mime, :string, 3
+      proto3_optional :magic, :string, 4
+      optional :index, :uint32, 8
+      optional :finish_reason, :enum, 9, "gooseai.FinishReason"
+      optional :seed, :uint32, 10
+      oneof :data do
+        optional :binary, :bytes, 5
+        optional :text, :string, 6
+        optional :tokens, :message, 7, "gooseai.Tokens"
+        optional :classifier, :message, 11, "gooseai.ClassifierParameters"
+      end
+    end
+    add_message "gooseai.PromptParameters" do
+      proto3_optional :init, :bool, 1
+      proto3_optional :weight, :float, 2
+    end
+    add_message "gooseai.Prompt" do
+      proto3_optional :parameters, :message, 1, "gooseai.PromptParameters"
+      oneof :prompt do
+        optional :text, :string, 2
+        optional :tokens, :message, 3, "gooseai.Tokens"
+        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
+      proto3_optional :latent_channels, :uint64, 3
+      proto3_optional :downsampling_factor, :uint64, 4
+      proto3_optional :cfg_scale, :float, 5
+    end
+    add_message "gooseai.ConditionerParameters" do
+      proto3_optional :vector_adjust_prior, :string, 1
+    end
+    add_message "gooseai.StepParameter" do
+      optional :scaled_step, :float, 1
+      proto3_optional :sampler, :message, 2, "gooseai.SamplerParameters"
+    end
+    add_message "gooseai.TransformType" do
+      oneof :type do
+        optional :diffusion, :enum, 1, "gooseai.DiffusionSampler"
+        optional :upscaler, :enum, 2, "gooseai.Upscaler"
+      end
+    end
+    add_message "gooseai.ImageParameters" do
+      proto3_optional :height, :uint64, 1
+      proto3_optional :width, :uint64, 2
+      repeated :seed, :uint32, 3
+      proto3_optional :samples, :uint64, 4
+      proto3_optional :steps, :uint64, 5
+      proto3_optional :transform, :message, 6, "gooseai.TransformType"
+      repeated :parameters, :message, 7, "gooseai.StepParameter"
+    end
+    add_message "gooseai.ClassifierConcept" do
+      optional :concept, :string, 1
+      proto3_optional :threshold, :float, 2
+    end
+    add_message "gooseai.ClassifierCategory" do
+      optional :name, :string, 1
+      repeated :concepts, :message, 2, "gooseai.ClassifierConcept"
+      proto3_optional :adjustment, :float, 3
+      proto3_optional :action, :enum, 4, "gooseai.Action"
+      proto3_optional :classifier_mode, :enum, 5, "gooseai.ClassifierMode"
+    end
+    add_message "gooseai.ClassifierParameters" do
+      repeated :categories, :message, 1, "gooseai.ClassifierCategory"
+      repeated :exceeds, :message, 2, "gooseai.ClassifierCategory"
+      proto3_optional :realized_action, :enum, 3, "gooseai.Action"
+    end
+    add_message "gooseai.Request" do
+      optional :engine_id, :string, 1
+      optional :request_id, :string, 2
+      optional :requested_type, :enum, 3, "gooseai.ArtifactType"
+      repeated :prompt, :message, 4, "gooseai.Prompt"
+      proto3_optional :conditioner, :message, 6, "gooseai.ConditionerParameters"
+      proto3_optional :classifier, :message, 7, "gooseai.ClassifierParameters"
+      oneof :params do
+        optional :image, :message, 5, "gooseai.ImageParameters"
+      end
+    end
+    add_enum "gooseai.FinishReason" do
+      value :NULL, 0
+      value :LENGTH, 1
+      value :STOP, 2
+      value :ERROR, 3
+      value :FILTER, 4
+    end
+    add_enum "gooseai.ArtifactType" do
+      value :ARTIFACT_NONE, 0
+      value :ARTIFACT_IMAGE, 1
+      value :ARTIFACT_VIDEO, 2
+      value :ARTIFACT_TEXT, 3
+      value :ARTIFACT_TOKENS, 4
+      value :ARTIFACT_EMBEDDING, 5
+      value :ARTIFACT_CLASSIFICATIONS, 6
+    end
+    add_enum "gooseai.DiffusionSampler" do
+      value :SAMPLER_DDIM, 0
+      value :SAMPLER_DDPM, 1
+      value :SAMPLER_K_EULER, 2
+      value :SAMPLER_K_EULER_ANCESTRAL, 3
+      value :SAMPLER_K_HEUN, 4
+      value :SAMPLER_K_DPM_2, 5
+      value :SAMPLER_K_DPM_2_ANCESTRAL, 6
+      value :SAMPLER_K_LMS, 7
+    end
+    add_enum "gooseai.Upscaler" do
+      value :UPSCALER_RGB, 0
+      value :UPSCALER_GFPGAN, 1
+      value :UPSCALER_ESRGAN, 2
+    end
+    add_enum "gooseai.Action" do
+      value :ACTION_PASSTHROUGH, 0
+      value :ACTION_REGENERATE_DUPLICATE, 1
+      value :ACTION_REGENERATE, 2
+      value :ACTION_OBFUSCATE_DUPLICATE, 3
+      value :ACTION_OBFUSCATE, 4
+      value :ACTION_DISCARD, 5
+    end
+    add_enum "gooseai.ClassifierMode" do
+      value :CLSFR_MODE_ZEROSHOT, 0
+      value :CLSFR_MODE_MULTICLASS, 1
+    end
+  end
+end
+
+module Gooseai
+  Token = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Token").msgclass
+  Tokens = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Tokens").msgclass
+  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
+  StepParameter = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.StepParameter").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
+  Request = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Request").msgclass
+  FinishReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.FinishReason").enummodule
+  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
+  Action = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.Action").enummodule
+  ClassifierMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ClassifierMode").enummodule
+end

data/lib/generation_services_pb.rb

@@ -0,0 +1,22 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# Source: generation.proto for package 'gooseai'
+
+require 'grpc'
+require 'generation_pb'
+
+module Gooseai
+  module GenerationService
+    class Service
+
+      include ::GRPC::GenericService
+
+      self.marshal_class_method = :encode
+      self.unmarshal_class_method = :decode
+      self.service_name = 'gooseai.GenerationService'
+
+      rpc :Generate, ::Gooseai::Request, stream(::Gooseai::Answer)
+    end
+
+    Stub = Service.rpc_stub_class
+  end
+end

data/lib/stability_sdk/cli.rb

@@ -0,0 +1,34 @@
+module StabilitySDK
+  class CLI
+    def self.save_answer(answer, options, logger)
+      logger.debug "received answer: #{answer}"
+      return if options[:no_store]
+
+      answer.artifacts.each_with_index do |artifact, idx|
+        filename_base = "#{options[:prefix] || "generation"}-#{answer.request_id}-#{answer.answer_id}-#{idx}"
+
+        filename = ""
+        contents = ""
+
+        case artifact.type
+        when :ARTIFACT_IMAGE
+          ext = MIME::Types[artifact.mime].first.preferred_extension
+          filename = "#{filename_base}.#{ext}"
+          contents = artifact.binary
+        when :ARTIFACT_CLASSIFICATIONS
+          ext = "pb.json"
+          filename = "#{filename_base}.#{ext}"
+          contents = artifact.classifier.to_json
+        else
+          logger.warn "not implemented for ArtifactType #{artifact.type}"
+        end
+
+        next if filename == "" || contents == ""
+
+        File.open(filename, "wb") do |f|
+          f.write(contents)
+        end
+      end
+    end
+  end
+end

data/lib/stability_sdk/client.rb

@@ -0,0 +1,75 @@
+require "grpc"
+require "generation_services_pb"
+
+module StabilitySDK
+  class Client
+    DEFAULT_API_HOST = "grpc.stability.ai:443"
+    DEFAULT_IMAGE_WIDTH = 512
+    DEFAULT_IMAGE_HEIGHT = 512
+    DEFAULT_SAMPLE_SIZE = 1
+    DEFAULT_STEPS = 50
+    DEFAULT_ENGINE_ID = "stable-diffusion-v1"
+    DEFAULT_CFG_SCALE = 7.0
+    DEFAULT_SAMPLER_ALGORITHM = Gooseai::DiffusionSampler::SAMPLER_K_LMS
+
+    sampler_algorithms = {
+      "ddim": Gooseai::DiffusionSampler::SAMPLER_DDIM,
+      "plms": Gooseai::DiffusionSampler::SAMPLER_DDPM,
+      "k_euler": Gooseai::DiffusionSampler::SAMPLER_K_EULER,
+      "k_euler_ancestral": Gooseai::DiffusionSampler::SAMPLER_K_EULER_ANCESTRAL,
+      "k_heun": Gooseai::DiffusionSampler::SAMPLER_K_HEUN,
+      "k_dpm_2": Gooseai::DiffusionSampler::SAMPLER_K_DPM_2,
+      "k_dpm_2_ancestral": Gooseai::DiffusionSampler::SAMPLER_K_DPM_2_ANCESTRAL,
+      "k_lms": Gooseai::DiffusionSampler::SAMPLER_K_LMS,
+    }
+
+    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 = Gooseai::GenerationService::Stub.new(host, creds)
+    end
+
+    def generate(prompt, options, &block)
+      image_param = image_param(options)
+      req = Gooseai::Request.new(
+        prompt: [Gooseai::Prompt.new(text: prompt)],
+        engine_id: options[:engine_id] || DEFAULT_ENGINE_ID,
+        image: image_param
+      )
+
+      @stub.generate(req).each do |answer|
+        block.call(answer)
+      end
+    end
+
+    def image_param(options={})
+      width = options.has_key?(:width) ? options[:width].to_i : DEFAULT_IMAGE_WIDTH
+      height = options.has_key?(:height) ? options[:height] : DEFAULT_IMAGE_HEIGHT
+      samples = options.has_key?(:num_samples) ? [: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]] : [rand(4294967295)]
+      transform = Gooseai::TransformType.new(
+        diffusion: options.has_key?(:sampler) ? sampler_algorithms[options[:sampler]] : DEFAULT_SAMPLER_ALGORITHM,
+      )
+      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,
+        ),
+      )]
+
+      return Gooseai::ImageParameters.new(
+        width: width,
+        height: height,
+        samples: samples,
+        steps: steps,
+        seed: seed,
+        transform: transform,
+        parameters: parameters,
+      )
+    end
+  end
+end

data/lib/stability_sdk/version.rb

@@ -0,0 +1,3 @@
+module StabilitySDK
+  VERSION = "0.1.0"
+end

data/lib/stability_sdk.rb

@@ -0,0 +1,8 @@
+require "stability_sdk/version"
+require "stability_sdk/client"
+require "stability_sdk/cli"
+
+module StabilitySDK
+  class Error < StandardError; end
+  class InsufficientParameter < StandardError; end
+end

data/proto/generation.proto

@@ -0,0 +1,179 @@
+syntax = 'proto3';
+package gooseai;
+option go_package = "./;generation";
+
+enum FinishReason {
+  NULL = 0;
+  LENGTH = 1;
+  STOP = 2;
+  ERROR = 3;
+  FILTER = 4;
+}
+
+
+enum ArtifactType {
+  ARTIFACT_NONE = 0;
+  ARTIFACT_IMAGE = 1;
+  ARTIFACT_VIDEO = 2;
+  ARTIFACT_TEXT = 3;
+  ARTIFACT_TOKENS = 4;
+  ARTIFACT_EMBEDDING = 5;
+  ARTIFACT_CLASSIFICATIONS = 6;
+}
+
+message Token {
+  optional string text = 1;
+  uint32 id = 2;
+}
+
+message Tokens {
+  repeated Token tokens = 1;
+  optional string tokenizer_id = 2;
+}
+
+message Artifact {
+  uint64 id = 1;
+  ArtifactType type = 2;
+  string mime = 3;
+  optional string magic = 4;
+  oneof data {
+    bytes binary = 5;
+    string text = 6;
+    Tokens tokens = 7;
+    ClassifierParameters classifier = 11;
+  }
+  uint32 index = 8;
+  FinishReason finish_reason = 9;
+  uint32 seed = 10;
+}
+
+message PromptParameters {
+  optional bool init = 1;
+  optional float weight = 2;
+}
+
+message Prompt {
+  optional PromptParameters parameters = 1;
+  oneof prompt {
+    string text = 2;
+    Tokens tokens = 3;
+    Artifact artifact = 4;
+  }
+}
+
+message AnswerMeta {
+  optional string gpu_id = 1;
+  optional string cpu_id = 2;
+  optional string node_id = 3;
+  optional string engine_id = 4;
+}
+
+message Answer {
+  string answer_id = 1;
+  string request_id = 2;
+  uint64 received = 3;
+  uint64 created = 4;
+  optional AnswerMeta meta = 6;
+  repeated Artifact artifacts = 7;
+}
+
+enum DiffusionSampler {
+  SAMPLER_DDIM = 0;
+  SAMPLER_DDPM = 1;
+  SAMPLER_K_EULER = 2;
+  SAMPLER_K_EULER_ANCESTRAL = 3;
+  SAMPLER_K_HEUN = 4;
+  SAMPLER_K_DPM_2 = 5;
+  SAMPLER_K_DPM_2_ANCESTRAL = 6;
+  SAMPLER_K_LMS = 7;
+}
+
+message SamplerParameters {
+  optional float eta = 1;
+  optional uint64 sampling_steps = 2;
+  optional uint64 latent_channels = 3;
+  optional uint64 downsampling_factor = 4;
+  optional float cfg_scale = 5;
+}
+
+message ConditionerParameters {
+  optional string vector_adjust_prior = 1;
+}
+
+enum Upscaler {
+  UPSCALER_RGB = 0;
+  UPSCALER_GFPGAN = 1;
+  UPSCALER_ESRGAN = 2;
+}
+
+message StepParameter {
+  float scaled_step = 1;
+  optional SamplerParameters sampler = 2;
+}
+
+message TransformType {
+  oneof type {
+      DiffusionSampler diffusion = 1;
+      Upscaler upscaler = 2;
+  }
+}
+
+message ImageParameters {
+  optional uint64 height = 1;
+  optional uint64 width = 2;
+  repeated uint32 seed = 3;
+  optional uint64 samples = 4;
+  optional uint64 steps = 5;
+  optional TransformType transform = 6;
+  repeated StepParameter parameters = 7;
+}
+
+enum Action {
+  ACTION_PASSTHROUGH = 0;
+  ACTION_REGENERATE_DUPLICATE = 1;
+  ACTION_REGENERATE = 2;
+  ACTION_OBFUSCATE_DUPLICATE = 3;
+  ACTION_OBFUSCATE = 4;
+  ACTION_DISCARD = 5;
+}
+
+enum ClassifierMode {
+  CLSFR_MODE_ZEROSHOT = 0;
+  CLSFR_MODE_MULTICLASS = 1;
+  /*CLSFR_MODE_ODDSRATIO = 2;*/
+}
+
+message ClassifierConcept {
+  string concept = 1;
+  optional float threshold = 2;
+}
+
+message ClassifierCategory {
+  string name = 1;
+  repeated ClassifierConcept concepts = 2;
+  optional float adjustment = 3;
+  optional Action action = 4;
+  optional ClassifierMode classifier_mode = 5;
+}
+
+message ClassifierParameters {
+  repeated ClassifierCategory categories = 1;
+  repeated ClassifierCategory exceeds = 2;
+  optional Action realized_action = 3;
+}
+
+message Request {
+  string engine_id = 1;
+  string request_id = 2;
+  ArtifactType requested_type = 3;
+  repeated Prompt prompt = 4;
+  oneof params {
+    ImageParameters image = 5;
+  }
+  optional ConditionerParameters conditioner = 6;
+  optional ClassifierParameters classifier = 7;
+}
+
+service GenerationService {
+  rpc Generate (Request) returns (stream Answer) {};
+}

data/stability_sdk.gemspec

@@ -0,0 +1,30 @@
+require_relative 'lib/stability_sdk/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "stability_sdk"
+  spec.version       = StabilitySDK::VERSION
+  spec.authors       = ["Kosei Moriyama"]
+  spec.email         = ["cou929@gmail.com"]
+
+  spec.summary       = "Ruby client for interacting with stability.ai APIs (e.g. stable diffusion inference)"
+  spec.description   = "Ruby client of https://github.com/Stability-AI/stability-sdk"
+  spec.homepage      = "https://github.com/cou929/stability-sdk-ruby"
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.3.0")
+
+  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"
+
+  # 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.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "grpc"
+  spec.add_dependency "mime-types"
+  spec.add_development_dependency "grpc-tools"
+end

metadata

@@ -0,0 +1,106 @@
+--- !ruby/object:Gem::Specification
+name: stability_sdk
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Kosei Moriyama
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2022-09-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: grpc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: mime-types
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: grpc-tools
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby client of https://github.com/Stability-AI/stability-sdk
+email:
+- cou929@gmail.com
+executables:
+- stability-client
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- exe/stability-client
+- 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/version.rb
+- proto/generation.proto
+- stability_sdk.gemspec
+homepage: https://github.com/cou929/stability-sdk-ruby
+licenses: []
+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
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.4
+signing_key: 
+specification_version: 4
+summary: Ruby client for interacting with stability.ai APIs (e.g. stable diffusion
+  inference)
+test_files: []