Stable Diffusion v1.4 · Ilustraciones de libros antiguos
Finetuning de CompVis/stable-diffusion-v1-4
sobre el dataset gigant/oldbookillustrations.
Solo se ha finetuneado la UNet; el VAE y el text encoder se mantienen congelados.
Uso
from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained("Joakamakaka/finetuned-sd-1.4-oldbookillustrations")
pipe = pipe.to("cpu")
prompt = "A sailing ship with billowing sails rides the tall waves of a stormy sea, old book illustration, engraving, antique"
image = pipe(prompt).images[0]
image.save("output.png")
Detalles del entrenamiento
| Parametro | Valor |
|---|---|
| Modelo base | CompVis/stable-diffusion-v1-4 |
| Dataset | gigant/oldbookillustrations |
| Columnas | 1600px / info_alt |
| Epocas | 2 |
| Learning rate | 1e-05 |
| Batch size | 2 |
| Resolucion | 512x512 |
| Trigger de estilo | old book illustration, engraving, antique |