stability_sdk 0.3.0 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3e707d41b381676dfbc958975bae028b82a52a7828529c52db7809db125791e9
-  data.tar.gz: 9e7aafbd1c0e5638ad2bc860312e55a0b2973ec3c8423fa050730cfa15887efa
+  metadata.gz: 4bfbab3c7b0c366f01681d5adf54193d9473741c8616d3417628011f87e60bea
+  data.tar.gz: f30748cc3ddefc1250b286f695cf039d8297970ef8757720620b9cec9d60e0cc
 SHA512:
-  metadata.gz: 37cb8a18a1accb3cf30f6dc780781f71dfe75ad648713a2c335d1b2be41f7546c0450d88ec0ae833354db24a3d3659fe0ac8163d1f3a1a076f08e8e344282c9c
-  data.tar.gz: aa66ce05d53d3d5bd6aa661f752a5d71f61c9835804d69f38834e07bc4e5259d971816eb4d4061264cfadc9a825a425b0469c794369eb48c1080cdbf1d1342d3
+  metadata.gz: 7f1eab7a664df5a09cd7785f9636e7e7542b87ed3ce475ceb5dd40005ec6ddc2e246346e258225d2e6f04b3472f28a316c91a32b3aa3e069acf031df5db20b82
+  data.tar.gz: c74dffd4dcbffb5d3211d6bc4762d76384badef93c85e33432c93bf7c3b4c1f0eab2b934c8af0f4065f791e1fd3204408cdc50a5c16037c3cc26b44ef6bc449e

data/README.md

@@ -148,9 +148,13 @@ cd api-interfaces
 
 # checkout some branch/commit you need
 git fetch
-git reset --hard origin/some_branch
+git checkout origin/some_branch # or `git reset --hard origin/some_branch`
 
 cd ..
+git add
+
+# if you need to follow the updates of submodules inside the Stability-AI/api-interfaces, you have to update submodules recursively
+git submodule update --init --recursive
 ```
 
 - build

data/Rakefile

@@ -11,7 +11,7 @@ task :default => :test
 
 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"
+  sh "grpc_tools_ruby_protoc -I api-interfaces/src/proto/ -I api-interfaces/src/tensorizer/proto/ --ruby_out=lib --grpc_out=lib api-interfaces/src/proto/generation.proto"
 end
 
 desc "Compile a dashboard proto file"

data/lib/generation_pb.rb

@@ -3,6 +3,8 @@
 
 require 'google/protobuf'
 
+require 'tensors_pb'
+
 Google::Protobuf::DescriptorPool.generated_pool.build do
   add_file("generation.proto", :syntax => :proto3) do
     add_message "gooseai.Token" do
@@ -28,6 +30,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
         optional :text, :string, 6
         optional :tokens, :message, 7, "gooseai.Tokens"
         optional :classifier, :message, 11, "gooseai.ClassifierParameters"
+        optional :tensor, :message, 14, "tensors.Tensor"
       end
     end
     add_message "gooseai.PromptParameters" do
@@ -48,6 +51,7 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       proto3_optional :latent_channels, :uint64, 3
       proto3_optional :downsampling_factor, :uint64, 4
       proto3_optional :cfg_scale, :float, 5
+      proto3_optional :init_noise_scale, :float, 6
     end
     add_message "gooseai.ConditionerParameters" do
       proto3_optional :vector_adjust_prior, :string, 1
@@ -108,6 +112,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       proto3_optional :steps, :uint64, 5
       proto3_optional :transform, :message, 6, "gooseai.TransformType"
       repeated :parameters, :message, 7, "gooseai.StepParameter"
+      proto3_optional :masked_area_init, :enum, 8, "gooseai.MaskedAreaInit"
+      proto3_optional :weight_method, :enum, 9, "gooseai.WeightMethod"
     end
     add_message "gooseai.ClassifierConcept" do
       optional :concept, :string, 1
@@ -186,6 +192,17 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :ARTIFACT_EMBEDDING, 5
       value :ARTIFACT_CLASSIFICATIONS, 6
       value :ARTIFACT_MASK, 7
+      value :ARTIFACT_LATENT, 8
+      value :ARTIFACT_TENSOR, 9
+    end
+    add_enum "gooseai.MaskedAreaInit" do
+      value :MASKED_AREA_INIT_ZERO, 0
+      value :MASKED_AREA_INIT_RANDOM, 1
+      value :MASKED_AREA_INIT_ORIGINAL, 2
+    end
+    add_enum "gooseai.WeightMethod" do
+      value :TEXT_ENCODER, 0
+      value :CROSS_ATTENTION, 1
     end
     add_enum "gooseai.DiffusionSampler" do
       value :SAMPLER_DDIM, 0
@@ -196,6 +213,8 @@ Google::Protobuf::DescriptorPool.generated_pool.build do
       value :SAMPLER_K_DPM_2, 5
       value :SAMPLER_K_DPM_2_ANCESTRAL, 6
       value :SAMPLER_K_LMS, 7
+      value :SAMPLER_K_DPMPP_2S_ANCESTRAL, 8
+      value :SAMPLER_K_DPMPP_2M, 9
     end
     add_enum "gooseai.Upscaler" do
       value :UPSCALER_RGB, 0
@@ -278,6 +297,8 @@ module Gooseai
   ChainRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ChainRequest").msgclass
   FinishReason = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.FinishReason").enummodule
   ArtifactType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.ArtifactType").enummodule
+  MaskedAreaInit = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.MaskedAreaInit").enummodule
+  WeightMethod = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("gooseai.WeightMethod").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

data/lib/stability_sdk/version.rb

@@ -1,3 +1,3 @@
 module StabilitySDK
-  VERSION = "0.3.0"
+  VERSION = "0.3.2"
 end

data/lib/tensors_pb.rb

@@ -0,0 +1,63 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensors.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensors.proto", :syntax => :proto3) do
+    add_message "tensors.Tensor" do
+      optional :dtype, :enum, 1, "tensors.Dtype"
+      repeated :shape, :int64, 2
+      optional :data, :bytes, 3
+      proto3_optional :attr_type, :enum, 4, "tensors.AttributeType"
+    end
+    add_message "tensors.Attribute" do
+      optional :name, :string, 1
+      oneof :value do
+        optional :module, :message, 3, "tensors.Module"
+        optional :tensor, :message, 4, "tensors.Tensor"
+        optional :string, :string, 5
+        optional :int64, :int64, 6
+        optional :float, :float, 7
+        optional :bool, :bool, 8
+      end
+    end
+    add_message "tensors.Module" do
+      optional :name, :string, 1
+      repeated :names, :string, 2
+      repeated :attributes, :message, 3, "tensors.Attribute"
+    end
+    add_enum "tensors.Dtype" do
+      value :DT_INVALID, 0
+      value :DT_FLOAT32, 1
+      value :DT_FLOAT64, 2
+      value :DT_FLOAT16, 3
+      value :DT_BFLOAT16, 4
+      value :DT_COMPLEX32, 5
+      value :DT_COMPLEX64, 6
+      value :DT_COMPLEX128, 7
+      value :DT_UINT8, 8
+      value :DT_INT8, 9
+      value :DT_INT16, 10
+      value :DT_INT32, 11
+      value :DT_INT64, 12
+      value :DT_BOOL, 13
+      value :DT_QUINT8, 14
+      value :DT_QINT8, 15
+      value :DT_QINT32, 16
+      value :DT_QUINT4_2, 17
+    end
+    add_enum "tensors.AttributeType" do
+      value :AT_PARAMETER, 0
+      value :AT_BUFFER, 1
+    end
+  end
+end
+
+module Tensors
+  Tensor = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensors.Tensor").msgclass
+  Attribute = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensors.Attribute").msgclass
+  Module = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensors.Module").msgclass
+  Dtype = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensors.Dtype").enummodule
+  AttributeType = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensors.AttributeType").enummodule
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stability_sdk
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.2
 platform: ruby
 authors:
 - Kosei Moriyama
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2022-11-17 00:00:00.000000000 Z
+date: 2022-12-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grpc
@@ -116,6 +116,7 @@ files:
 - lib/stability_sdk/client.rb
 - lib/stability_sdk/dashboard_client.rb
 - lib/stability_sdk/version.rb
+- lib/tensors_pb.rb
 - stability_sdk.gemspec
 homepage: https://github.com/cou929/stability-sdk-ruby
 licenses: