init

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image
+from diffusers.utils import load_image
+import torch
+
+if torch.cuda.is_available():
+  device = "cuda"
+elif torch.backends.mps.is_available():
+  device = "mps"
+else:
+  device = "cpu"
+
+
+pipes = {
+  "txt2img": AutoPipelineForText2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16").to(device),
+  "img2img": AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16").to(device)
+}
+
+
+def run(prompt, image):
+  print(f"prompt={prompt}, image={image}")
+  if image is None:
+    return pipes["txt2img"](prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]
+  else:
+    image = image.resize((512,512))
+    print(f"img2img image={image}")
+    return pipes["img2img"](prompt, image=image, num_inference_steps=2, strength=0.5, guidance_scale=0.0).images[0]
+
+demo = gr.Interface(
+    run,
+    inputs=[
+      gr.Textbox(label="Prompt"),
+      gr.Image(type="pil")
+    ],
+    outputs=gr.Image(width=512,height=512),
+    live=True
+)
+#demo.dependencies[0]["show_progress"] = "minimal"
+demo.launch()

install.json

@@ -0,0 +1,25 @@
+{
+  "requires": [{
+    "type": "conda",
+    "name": "ffmpeg",
+    "args": "-c conda-forge"
+  }],
+  "run": [{
+    "method": "shell.run",
+    "params": {
+      "venv": "env",
+      "message": "pip install -r requirements.txt"
+    }
+  }, {
+    "method": "input",
+    "params": {
+      "title": "Install Success",
+      "description": "Go back to the dashboard and launch the app!"
+    }
+  }, {
+    "method": "browser.open",
+    "params": {
+      "uri": "/?selected=sdxl turbo"
+    }
+  }]
+}

pinokio.js

@@ -0,0 +1,29 @@
+module.exports = {
+  title: "sdxl turbo",
+  menu: async (kernel) => {
+    let installed = await kernel.exists(__dirname, "env")
+    if (installed) {
+      let session = await kernel.require(__dirname, "session.json")
+      console.log("session", session)
+      return [{
+        when: "start.json",
+        on: "<i class='fa-solid fa-spin fa-circle-notch'></i> Running",
+        type: "label"
+      }, {
+        when: "start.json",
+        off: "<i class='fa-solid fa-power-off'></i> Start",
+        href: "start.json?fullscreen=true&run=true",
+      }, {
+        when: "start.json",
+        on: "<i class='fa-solid fa-rocket'></i> Launch",
+        href: (session && session.url ? session.url : "http://127.0.0.1:7860"),
+        target: "_blank"
+      }]
+    } else {
+      return [{
+        html: '<i class="fa-solid fa-plug"></i> Install',
+        href: "install.json?run=true&fullscreen=true"
+      }]
+    }
+  }
+}

requirements.txt

@@ -0,0 +1,4 @@
+diffusers
+transformers
+accelerate
+gradio

start.json

@@ -0,0 +1,23 @@
+{
+  "daemon": true,
+  "run": [{
+    "method": "shell.run",
+    "params": {
+      "venv": "env",
+      "message": "python app.py",
+      "on": [{ "event": "/http:\/\/[0-9.:]+/", "done": true }]
+    }
+  }, {
+    "method": "self.set",
+    "params": {
+      "session.json": {
+        "url": "{{input.event[0]}}"
+      }
+    }
+  }, {
+    "method": "browser.open",
+    "params": {
+      "uri": "{{self.session.url}}"
+    }
+  }]
+}