Merge pull request #43 from KevinBatdorf/stable-diffusion-v2

Add stable diffusion v2 test version
KevinBatdorf · Nov 25, 2022 · a3c84ef · a3c84ef
2 parents 20cae07 + e716975
commit a3c84ef
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 7 deletions.
diff --git a/.github/workflows/build-production-zip.yml b/.github/workflows/build-production-zip.yml
@@ -15,7 +15,7 @@ jobs:
                   CI: true
 
             - name: Package
-              uses: actions/upload-artifact@v2
+              uses: actions/upload-artifact@v3
               with:
                   name: stable-diffusion
                   retention-days: 5

diff --git a/cypress/support/gutenberg.js b/cypress/support/gutenberg.js
@@ -69,7 +69,13 @@ export const closeBlockInserter = () => {
 };
 export const addBlock = (slug) => {
     cy.openBlockInserter();
-    cy.get(`button[class*="${slug}"]`).click();
+    cy.window().then((win) => {
+        cy.waitUntil(() =>
+            win.document.querySelector(`button[class*="${slug}"]`),
+        );
+        cy.get(`button[class*="${slug}"]`).should('exist');
+        cy.get(`button[class*="${slug}"]`).click({ force: true });
+    });
 };
 export const wpDataSelect = (store, selector, ...parameters) => {
     cy.window().then((win) => {

diff --git a/readme.txt b/readme.txt
@@ -2,7 +2,7 @@
 Contributors:      kbat82
 Tags:              block, stable diffusion, ai, prompt, artificial intelligence, generate, pokemon, dall-e, midjourney
 Tested up to:      6.1
-Stable tag:        0.6.0
+Stable tag:        0.7.0
 License:           GPL-2.0-or-later
 License URI:       https://www.gnu.org/licenses/gpl-2.0.html
 
@@ -56,6 +56,9 @@ We also offer additional features on top of the Replicate API, such as prompt/ou
 
 == Changelog ==
 
+= 0.7.0 - 2022-11-24 =
+- Feature: Add test version of Stable Diffusion v2
+
 = 0.6.0 - 2022-11-22 =
 - Feature: Add OpenJourney model - Stable Diffusion fine tuned on Midjourney v4 images.
 

diff --git a/src/components/features/PromptGenerator.tsx b/src/components/features/PromptGenerator.tsx
@@ -27,9 +27,11 @@ export const PromptGenerator = ({
 
     // For now only show on this model
     if (
-        !['stability-ai/stable-diffusion', 'prompthero/openjourney'].includes(
-            currentModel,
-        )
+        ![
+            'stability-ai/stable-diffusion',
+            'prompthero/openjourney',
+            'cjwbw/stable-diffusion-v2',
+        ].includes(currentModel)
     )
         return null;
 

diff --git a/src/models.ts b/src/models.ts
@@ -11,6 +11,15 @@ export const models: ModelAttributes[] = [
         ),
         image: 'https://bucketeer-be99e627-94e7-4e5b-a292-54eeb40ac303.s3.amazonaws.com/public/models_models_featured_image/c63d605d-b5e3-4fb0-9b0d-79f00a710f7c/out-0.png',
     },
+    {
+        id: 'cjwbw/stable-diffusion-v2',
+        name: __('Stable Diffusion v2 (Test)', 'stable-diffusion'),
+        description: __(
+            'Test implimentation of Stable Diffusion V2',
+            'stable-diffusion',
+        ),
+        image: 'https://bucketeer-be99e627-94e7-4e5b-a292-54eeb40ac303.s3.amazonaws.com/public/models_models_featured_image/d8e0c1ca-1256-43d2-b5e6-773c61ffc629/out-0_2.png',
+    },
     {
         id: 'lambdal/text-to-pokemon',
         name: __('Text to Pokémon', 'stable-diffusion'),

diff --git a/src/types.ts b/src/types.ts
@@ -1,5 +1,6 @@
 export type AvailableModels =
     | 'stability-ai/stable-diffusion'
+    | 'cjwbw/stable-diffusion-v2'
     | 'lambdal/text-to-pokemon'
     | 'prompthero/openjourney'
     | 'methexis-inc/img2prompt'

diff --git a/stable-diffusion.php b/stable-diffusion.php
@@ -4,7 +4,7 @@
  * Description:       Generate unique images from text prompts using machine learning, all in the cloud.
  * Requires at least: 5.8
  * Requires PHP:      7.0
- * Version:           0.6.0
+ * Version:           0.7.0
  * Author:            Kevin Batdorf
  * License:           GPL-2.0-or-later
  * License URI:       https://www.gnu.org/licenses/gpl-2.0.html