chore: Remove dump example and add sd-gpu example

examples/stable-diffusion/README.md

@@ -0,0 +1,15 @@
+# Stable Diffusion
+
+## Requirements
+
+<ol>
+  <li><a href="https://huggingface.co/join">Sign up to Huggingface</a></li>
+  <li><a href="https://huggingface.co/CompVis/stable-diffusion-v1-4">Accept the Stable Diffusion models agreement</a></li>
+  <li><a href="https://huggingface.co/settings/tokens">Create an Access Token</a>. You’ll use it in the Python script below.</li>
+</ol>
+
+## Steps
+
+- `envd up`
+- `export HUGGINGFACE_TOKEN=<token>
+- `python main.py`

examples/stable-diffusion/build.envd

@@ -0,0 +1,14 @@
+def build():
+    base(os="ubuntu20.04", language="python")
+    # config.pip_index(url = "https://pypi.tuna.tsinghua.edu.cn/simple")
+    install.python_packages(
+        [
+            "torch",
+            "transformers",
+            "diffusers",
+        ]
+    )
+    install.cuda(version="11.2.0", cudnn="8")
+    runtime.mount(
+        host_path="~/.cache/huggingface", envd_path="/home/envd/.cache/huggingface"
+    )

examples/stable-diffusion/main.py

@@ -0,0 +1,27 @@
+import random
+import sys
+import os
+
+from diffusers import StableDiffusionPipeline
+
+device = "cuda"
+
+
+def dummy(images, **kwargs):
+    return images, False
+
+
+# Read prompt from command line
+prompt = " ".join(sys.argv[1:])
+
+pipe = StableDiffusionPipeline.from_pretrained(
+    "CompVis/stable-diffusion-v1-4", use_auth_token=os.environ["HUGGINGFACE_TOKEN"]
+)
+pipe.to(device)
+pipe.safety_checker = dummy
+
+# Run until we exit with CTRL+C
+while True:
+    n = random.randint(1000, 9999)
+    image = pipe(prompt, guidance_scale=7.5)["sample"][0]
+    image.save(f"{n}.jpeg")