stability_sdk 0.2.9 → 0.2.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 27bb9b00352e1cf979dedd17438a5dc2a3dae93ed0d74a747e30f297152e81a2
-  data.tar.gz: c125901e07cc2f6d0704a423e2dad2541c3e31efbeb1ff69993f65bf804dcabb
+  metadata.gz: 0b45f00aa32974424860512f59fb3c1745b6c3b8b08f359e95ccd603f637642d
+  data.tar.gz: a67397bdc2b10b0bc49b7a508e3283c20fa4d7ea4ae4b4598b086b0ecc43b323
 SHA512:
-  metadata.gz: 4b79d8290c5c958f10a00f103efe6959b8e1f408d8ca3393453afb0a7433eda250252edab93195a1dd7885b9a3860aeffccd27833289f38ea373ff3bf1a274f0
-  data.tar.gz: 4481b59d12990bf00e5b716611dbd8055df7d4d262a4f1647ea2b217c9122cd790b671e76d51f1c082e83f2a0cebb1d62e2a80d7cb08f1ae96755db299a89dcb
+  metadata.gz: f2fe617756b1d720e1d3a5c63807a7dbde52b70df2eb7a3c11a2bde803245f0e00c25b29d0d1893359de2243d333cf82ce274e23e03f220dadcad1d6f661261c
+  data.tar.gz: e50e38e2260d9e376581c03555468273c4378dea33cefdce1fb3e00e0d32db8aa592a63ba0e87f9685d0c89cdbd95c75af5093fc3e724711d862727d346028ba

data/.gitignore

@@ -6,3 +6,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+Gemfile.lock
+.DS_Store

data/exe/stability-client

@@ -2,6 +2,7 @@
 
 require "optparse"
 require "mime/types"
+require "fastimage"
 require "logger"
 require "stability_sdk"
 
@@ -42,6 +43,11 @@ raise StabilitySDK::InsufficientParameter, "api key is required" if !options.has
 
 raise StabilitySDK::InsufficientParameter, "init_image is required if mask_image is provided" if options.has_key?(:mask_image) && !options.has_key?(:init_image)
 
+if options.has_key?(:init_image)
+  size = FastImage.size(options[: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.generate(prompt, options) do |answer|

data/lib/stability_sdk/client.rb

@@ -42,6 +42,11 @@ module StabilitySDK
     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
+
+      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)]

data/lib/stability_sdk/version.rb

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

data/lib/stability_sdk.rb

@@ -5,4 +5,5 @@ require "stability_sdk/cli"
 module StabilitySDK
   class Error < StandardError; end
   class InsufficientParameter < StandardError; end
+  class InvalidParameter < StandardError; end
 end

data/stability_sdk.gemspec

@@ -27,5 +27,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_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.9
+  version: 0.2.10
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-10-03 00:00:00.000000000 Z
+date: 2022-10-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -38,6 +38,26 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 3.0.0
+- !ruby/object:Gem::Dependency
+  name: fastimage
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.2'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.2.6
 - !ruby/object:Gem::Dependency
   name: grpc-tools
   requirement: !ruby/object:Gem::Requirement
@@ -65,7 +85,6 @@ files:
 - ".gitmodules"
 - ".travis.yml"
 - Gemfile
-- Gemfile.lock
 - LICENSE
 - README.md
 - Rakefile

data/Gemfile.lock

@@ -1,35 +0,0 @@
-PATH
-  remote: .
-  specs:
-    stability_sdk (0.2.9)
-      grpc (>= 1.41.1)
-      mime-types (>= 3.0.0)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    google-protobuf (3.21.7)
-    googleapis-common-protos-types (1.4.0)
-      google-protobuf (~> 3.14)
-    grpc (1.49.1)
-      google-protobuf (~> 3.21)
-      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
-  ruby
-
-DEPENDENCIES
-  grpc-tools
-  minitest (~> 5.0)
-  rake (~> 12.0)
-  stability_sdk!
-
-BUNDLED WITH
-   2.2.2