stable diffusion experiment

manufy · May 31, 2024 · 9948999 · 9948999
1 parent eee8477
commit 9948999
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/3-experiments/6-text-to-image/diffusers/stable-diffusion/generate.py b/3-experiments/6-text-to-image/diffusers/stable-diffusion/generate.py
@@ -0,0 +1,16 @@
+from diffusers import DiffusionPipeline
+import torch
+
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", 
+                                         torch_dtype=torch.float16,
+                                         use_safetensors=True,
+                                         variant="fp16",
+                                         cache_dir="./custom_cache")
+#∫pipe.to("cuda")
+
+# if using torch < 2.0
+# pipe.enable_xformers_memory_efficient_attention()
+
+prompt = "An astronaut riding a green horse"
+
+images = pipe(prompt=prompt).images[0]
diff --git a/3-experiments/6-text-to-image/diffusers/stable-diffusion/pipeline.py b/3-experiments/6-text-to-image/diffusers/stable-diffusion/pipeline.py
@@ -0,0 +1,8 @@
+from diffusers import DiffusionPipeline
+
+
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",  cache_dir="./custom_cache")
+
+prompt = "An astronaut riding a green horse"
+
+images = pipeline(prompt=prompt).images[0]
diff --git a/3-experiments/6-text-to-image/diffusers/stable-diffusion/requirements.txt b/3-experiments/6-text-to-image/diffusers/stable-diffusion/requirements.txt
@@ -0,0 +1,6 @@
+diffusers
+invisible_watermark
+transformers
+accelerate
+safetensors
+