Dipl0/pepe-diffuser

🐸 Pepe Diffuser

Stable Diffusion 2 fine-tuné sur des images de Pepe the Frog.

Démo : Pepe Diffuser Space

Utilisation

import torch
from diffusers import DPMSolverMultistepScheduler, StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("Dipl0/pepe-diffuser", torch_dtype=torch.float16)
pipe.scheduler = DPMSolverMultistepScheduler.from_config(
    pipe.scheduler.config, algorithm_type="dpmsolver++", use_karras_sigmas=True
)
pipe.to("cuda")

image = pipe("pepe the frog surfing on the moon", num_inference_steps=22, guidance_scale=7).images[0]
image.save("pepe.png")

Commencer le prompt par pepe the frog donne les meilleurs résultats.

Entraînement

• Modèle de base : stabilityai/stable-diffusion-2
• Données : Dipl0/pepe_image_stable_diff
• Script : train_text_to_image de diffusers (février 2023)

Limites

Projet perso et expérimental. Les images d'entraînement viennent de mèmes internet : le modèle peut reproduire leur humour et leurs clichés.