1from diffusers import StableDiffusionPipeline
2
3pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/miniSD-diffusers")
4pipe = pipe.to("cuda")
5
6prompt = "a photograph of an wrinkly old man laughing"
7image = pipe(prompt, width=256, height=256).images[0]
8image.save('test.jpg')
-
22,000 steps fine-tuning only the attention layers of the unet, learn rate=1e-5, batch size=256
-
66,000 steps training the full unet, learn rate=5e-5, batch size=552
-
GPUs provided by
Lambda GPU Cloud
-
-
This model is open access and available to all, with a CreativeML OpenRAIL-M license further specifying rights and usage. The CreativeML OpenRAIL License specifies: