stability_sdk 0.2.3 → 0.2.5
Sign up to get free protection for your applications and to get access to all the features.
- 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(
|