stability_sdk 0.2.1 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 401df5c1aa6c7c458a171cc6e7f437a420d56a17f394236c42afe6b7114986bc
-  data.tar.gz: 744723a641d79456d255a0120118f8d218137d0901c760ede079d9dc51301f50
+  metadata.gz: 355759d5ba4bcb0ba3b688e85cc87651303710b7f79be7c63233b87f55c76696
+  data.tar.gz: 6f216e291a32250132cc07ba6c9083eefa2927332c3fefefef19971f412d80c8
 SHA512:
-  metadata.gz: f37b8db84a96fb1df551ef789e43138eb80adb8e8540a7b3619e44b66bed7b193aeb704ab8099daaf64f761c3e92bfd90088faf3285c9a1ffc07a06e457ae20a
-  data.tar.gz: 7b8b4bec82c39a5046bbe805e213fe5f4c5ddb0dd279352bd619768a711fc17a0254a958ddeac46d3835a76e75e7578ab11e82fdda9cb77d477d62ba5677c222
+  metadata.gz: 8f36ca048b79f3f79723427b7613bba2af9cb2f992796a7354b2b74a0c76ff6c72419aabf9c5dd0f64d8ee32b240fb1795e8a7c55d2b67e2b7f0f380c46ca4ab
+  data.tar.gz: 4b1ddbfc50e40a39b201082a606a9d6960b6b93b6233fa031d7e2dfc961f8d31abca1a6b180aab84727ef4c04b43692233776dba84f0750e54c63128eb5e1515

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    stability_sdk (0.2.1)
+    stability_sdk (0.2.3)
       grpc (>= 1.0.0)
       mime-types (>= 3.0.0)
 

data/lib/stability_sdk/client.rb

@@ -1,5 +1,6 @@
 require "grpc"
 require "generation_services_pb"
+require "pry"
 
 module StabilitySDK
   class Client
@@ -31,7 +32,12 @@ module StabilitySDK
       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)
+      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::GenerationService::Stub.new(host, creds, **stub_params)
     end
 
     def generate(prompt, options, &block)
@@ -39,7 +45,7 @@ module StabilitySDK
       height = options.has_key?(:height) ? options[:height].to_i : 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)]
+      seed = options.has_key?(:seed) ? [options[:seed].to_i] : [rand(4294967295)]
       transform = Gooseai::TransformType.new(
         diffusion: options.has_key?(:sampler) ? sampler_algorithms[options[:sampler]] : DEFAULT_SAMPLER_ALGORITHM,
       )

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.2.1"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.3
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-09-09 00:00:00.000000000 Z
+date: 2022-09-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc