app.py

app.py

@@ -0,0 +1,24 @@
+#@title ##Run for generating images.
+
+prompt = "A very good dog by [yourname]" #@param {type:"string"}
+negative_prompt = "" #@param {type:"string"}
+num_samples = 4 #@param {type:"number"}
+guidance_scale = 7.5 #@param {type:"number"}
+num_inference_steps = 50 #@param {type:"number"}
+height = 512 #@param {type:"number"}
+width = 512 #@param {type:"number"}
+
+with autocast("cuda"), torch.inference_mode():
+    images = pipe(
+        prompt,
+        height=height,
+        width=width,
+        negative_prompt=negative_prompt,
+        num_images_per_prompt=num_samples,
+        num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale,
+        generator=g_cuda
+    ).images
+
+for img in images:
+    display(img)