stability_sdk 0.2.3 → 0.2.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/Gemfile.lock +1 -1
- data/README.md +2 -3
- data/lib/stability_sdk/client.rb +1 -2
- data/lib/stability_sdk/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 452e6a45ca110b36eb3d6fafa4201a1be7ad4b6e0b6f677fe37f6a66fc8f5749
|
|
4
|
+
data.tar.gz: 1b73ee6ab82c18f63e786d71b2eee472e0db32572c4444ef5b3d5ac92003e3e7
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 481f539dc5bc36e67ebb3d1a1ec759280e4ed8ef0b02c49f0dfc3d7ce657db7b295e26d28f69bf7a24a5f410cc74ac62e083b2eaed78b6db2a19853150f8cc47
|
|
7
|
+
data.tar.gz: b9cfebe4973e2919eec0ffa24aa3b23d6326d46e017be05250bcba6d12ea3a2e539014eed9ef172bb6b0fa1415ffd48823813413b49de6f73b4ac3b5acbb1781
|
data/Gemfile.lock
CHANGED
data/README.md
CHANGED
|
@@ -19,8 +19,7 @@ Or install it yourself as:
|
|
|
19
19
|
$ gem install stability_sdk
|
|
20
20
|
|
|
21
21
|
## Usage
|
|
22
|
-
|
|
23
|
-
First you need to create a [DreamStudio](https://beta.dreamstudio.ai/home)'s account and get an API Key of it.
|
|
22
|
+
First, you need to create a [DreamStudio](https://beta.dreamstudio.ai/home)'s account and get an API Key of it.
|
|
24
23
|
|
|
25
24
|
- Access [DreamStudio](https://beta.dreamstudio.ai/dream) and create an account if you have not had it
|
|
26
25
|
- Go to the [membership page](https://beta.dreamstudio.ai/membership)
|
|
@@ -72,7 +71,7 @@ This sample code saves a generated image as `result.png`.
|
|
|
72
71
|
```ruby
|
|
73
72
|
require "stability_sdk"
|
|
74
73
|
|
|
75
|
-
client = StabilitySDK::Client.new(api_key: "YOUR_API_KEY")
|
|
74
|
+
client = StabilitySDK::Client.new(api_key: "YOUR_API_KEY", timeout: 600)
|
|
76
75
|
|
|
77
76
|
prompt = "your prompot here"
|
|
78
77
|
options = {}
|
data/lib/stability_sdk/client.rb
CHANGED
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
require "grpc"
|
|
2
2
|
require "generation_services_pb"
|
|
3
|
-
require "pry"
|
|
4
3
|
|
|
5
4
|
module StabilitySDK
|
|
6
5
|
class Client
|
|
@@ -43,7 +42,7 @@ module StabilitySDK
|
|
|
43
42
|
def generate(prompt, options, &block)
|
|
44
43
|
width = options.has_key?(:width) ? options[:width].to_i : DEFAULT_IMAGE_WIDTH
|
|
45
44
|
height = options.has_key?(:height) ? options[:height].to_i : DEFAULT_IMAGE_HEIGHT
|
|
46
|
-
samples = options.has_key?(:num_samples) ? [:num_samples].to_i : DEFAULT_SAMPLE_SIZE
|
|
45
|
+
samples = options.has_key?(:num_samples) ? options[:num_samples].to_i : DEFAULT_SAMPLE_SIZE
|
|
47
46
|
steps = options.has_key?(:steps) ? options[:steps].to_i : DEFAULT_STEPS
|
|
48
47
|
seed = options.has_key?(:seed) ? [options[:seed].to_i] : [rand(4294967295)]
|
|
49
48
|
transform = Gooseai::TransformType.new(
|