模型:
lambdalabs/miniSD-diffusers
类库:
Diffusersfrom diffusers import StableDiffusionPipeline pipe = StableDiffusionPipeline.from_pretrained("lambdalabs/miniSD-diffusers") pipe = pipe.to("cuda") prompt = "a photograph of an wrinkly old man laughing" image = pipe(prompt, width=256, height=256).images[0] image.save('test.jpg')
Fine tuned from the stable-diffusion 1.4 checkpoint as follows:
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
Trained on LAION Improved Aesthetics 6plus .
Trained using https://github.com/justinpinkney/stable-diffusion , original checkpoint available here
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: