stability_sdk 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3e707d41b381676dfbc958975bae028b82a52a7828529c52db7809db125791e9
-  data.tar.gz: 9e7aafbd1c0e5638ad2bc860312e55a0b2973ec3c8423fa050730cfa15887efa
+  metadata.gz: eb4bcef5f20a0b7147814c3c3cd90df2b39a54dc162d207a6c7f91141888cdb3
+  data.tar.gz: 77aa1b30334169b9437ca8798334fb081b7ff1c4305d09c69128bc5348768ba6
 SHA512:
-  metadata.gz: 37cb8a18a1accb3cf30f6dc780781f71dfe75ad648713a2c335d1b2be41f7546c0450d88ec0ae833354db24a3d3659fe0ac8163d1f3a1a076f08e8e344282c9c
-  data.tar.gz: aa66ce05d53d3d5bd6aa661f752a5d71f61c9835804d69f38834e07bc4e5259d971816eb4d4061264cfadc9a825a425b0469c794369eb48c1080cdbf1d1342d3
+  metadata.gz: d82e9bd6fdc0cec8b74e6ef774a82bd0d922327d99adae5b8ecccb55e7c7b3111e59d99c46fba8576aacbfd65fd4b3ea0c69a6e691b4b9967319a67b8a778b3f
+  data.tar.gz: 11ea3112c72f6a4a76fa6cd21457bf3808fddfa6956ffa170e7eebfdbdaca58f2e91dc68accb813b91bda6d82874c73b5d3fc4cf59025acdf96b969f98305904

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.1"
 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.1
 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