replicate-ruby 0.2.0 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 82385c582239598bb4b62a3e133228b7e6fc417ea57a047c1489e5d94d787faf
-  data.tar.gz: 4ccecfdb33f1a4cf3218587c7ed4fa28475da1ca13369e59d5875b07e4f092f8
+  metadata.gz: f7a1858f78be5077c9396f73607b07e9cc08531eb82476f8dc017d2da3cbcf38
+  data.tar.gz: 0a405743c2405687c58a27d8b6ad1078958eb0c237c3ab1a89922c9a17f150f2
 SHA512:
-  metadata.gz: c8edef78b1309f5873997b33b1de5de82f2f8982ce2ec3ed32b0cdeca0e9038c77b4d56e6173441d106160e77fb1e0d2311024d5cae2c68a1fc91ef6c62cb725
-  data.tar.gz: 56958524475fbd6636e1eb3f409b4dad100fe8179a982a746a0ee2ea4356ffa66724269a9da0e226e374bbe34c2526aacf3a358e5847ed55429ebf06700f2383
+  metadata.gz: 5141a0361e88ffec603dd331c813b3c99bff3e3f14bab88eb528ae7e519c117e5104463b8d7067ca0736be6e06b15d6ef512b668aa500caae40a1d2a3080d7a6
+  data.tar.gz: 23abec8601415dce6a56ac284889f1cf1c97ef6aff834cd23c2fdb1429be016a4b9880b401169cd4a04ceb67c1d022f109673c251b5314e2150d61a1e39c4ac6

data/Gemfile.lock

@@ -1,61 +1,61 @@
 PATH
   remote: .
   specs:
-    replicate-ruby (0.1.7)
+    replicate-ruby (0.2.1)
       addressable
-      faraday (>= 2.0)
+      faraday (>= 1.0)
       faraday-multipart
       faraday-retry
 
 GEM
   remote: https://rubygems.org/
   specs:
-    addressable (2.8.1)
+    addressable (2.8.4)
       public_suffix (>= 2.0.2, < 6.0)
     ast (2.4.2)
     byebug (11.1.3)
     coderay (1.1.3)
     crack (0.4.5)
       rexml
-    faraday (2.7.2)
+    faraday (2.7.4)
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
     faraday-multipart (1.0.4)
       multipart-post (~> 2)
     faraday-net_http (3.0.2)
-    faraday-retry (2.0.0)
+    faraday-retry (2.1.0)
       faraday (~> 2.0)
     hashdiff (1.0.1)
-    json (2.6.2)
+    json (2.6.3)
     method_source (1.0.0)
-    minitest (5.16.3)
-    multipart-post (2.2.3)
-    parallel (1.22.1)
-    parser (3.1.2.1)
+    minitest (5.18.0)
+    multipart-post (2.3.0)
+    parallel (1.23.0)
+    parser (3.2.2.1)
       ast (~> 2.4.1)
-    pry (0.14.1)
+    pry (0.14.2)
       coderay (~> 1.1)
       method_source (~> 1.0)
-    public_suffix (5.0.0)
+    public_suffix (5.0.1)
     rainbow (3.1.1)
     rake (13.0.6)
-    regexp_parser (2.6.0)
+    regexp_parser (2.8.0)
     rexml (3.2.5)
-    rubocop (1.36.0)
+    rubocop (1.51.0)
       json (~> 2.3)
       parallel (~> 1.10)
-      parser (>= 3.1.2.1)
+      parser (>= 3.2.0.0)
       rainbow (>= 2.2.2, < 4.0)
       regexp_parser (>= 1.8, < 3.0)
       rexml (>= 3.2.5, < 4.0)
-      rubocop-ast (>= 1.20.1, < 2.0)
+      rubocop-ast (>= 1.28.0, < 2.0)
       ruby-progressbar (~> 1.7)
-      unicode-display_width (>= 1.4.0, < 3.0)
-    rubocop-ast (1.21.0)
-      parser (>= 3.1.1.0)
-    ruby-progressbar (1.11.0)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.28.1)
+      parser (>= 3.2.1.0)
+    ruby-progressbar (1.13.0)
     ruby2_keywords (0.0.5)
-    unicode-display_width (2.3.0)
+    unicode-display_width (2.4.2)
     webmock (3.18.1)
       addressable (>= 2.8.0)
       crack (>= 0.3.2)

data/README.md

@@ -60,14 +60,13 @@ There is support for the [experimental dreambooth endpoint](https://replicate.co
 
 First, upload your training dataset:
 
-```
-upload = Replicate.client.create_upload
-upload.attach('tmp/data.zip') # replace with the path to your zip file
+```ruby
+upload = Replicate.client.upload_zip('tmp/data.zip') # replace with the path to your zip file
 ```
 
 Then start training a new model using, for instance:
 
-```
+```ruby
 training = Replicate.client.create_training(
   input: {
     instance_prompt: "zwx style",
@@ -81,7 +80,7 @@ training = Replicate.client.create_training(
 
 As soon as the model has finished training, you can run predictions on it:
 
-```
+```ruby
 prediction = Replicate.client.create_prediction(
   input: {
     prompt: 'your prompt, zwx style'
@@ -90,6 +89,12 @@ prediction = Replicate.client.create_prediction(
 )
 ```
 
+You can also download the `output.zip` file from the dreambooth training prediction, unzip it, and then convert the trained model to a Stable Diffusion checkpoint with [convert_diffusers_to_sd.py](https://gist.github.com/jachiam/8a5c0b607e38fcc585168b90c686eb05).
+
+```bash
+python ./convert_diffusers_to_sd.py --model_path ~/Downloads/output --checkpoint_path ~/Downloads/output.ckpt
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake test` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/replicate/client/upload.rb

@@ -4,10 +4,18 @@ module Replicate
   class Client
     # Methods for the Prediction API
     module Upload
+      # Create upload object and upload zip file
+      def upload_zip(zip_path)
+        filename = zip_path.split('/')[-1]
+        upload = create_upload(filename)
+        upload.attach(zip_path)
+        upload
+      end
+
       # Create an upload
       # @see https://replicate.com/blog/dreambooth-api
-      def create_upload
-        response = dreambooth_endpoint.post("upload/data.zip")
+      def create_upload(filename = 'data.zip')
+        response = dreambooth_endpoint.post("upload/#{filename}")
         Replicate::Record::Upload.new(self, response)
       end
 

data/lib/replicate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Replicate
-  VERSION = "0.2.0"
+  VERSION = "0.2.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: replicate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
 platform: ruby
 authors:
 - Dreaming Tulpa
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-08 00:00:00.000000000 Z
+date: 2023-05-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '1.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: faraday-retry
   requirement: !ruby/object:Gem::Requirement
@@ -159,7 +159,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.6
+rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
 summary: Ruby client for Replicate