pepe-diffuser / README.md
Dipl0's picture
Update README.md
3d2c54b
metadata
library_name: diffusers
pipeline_tag: text-to-image
tags:
  - pepe

How to use

To prompt you can use the following code

  from diffusers import StableDiffusionPipeline
  
  model_path = "Dipl0/pepe-diffuser"
  pipe = StableDiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
  pipe.to("cuda")
  
  prompt = "pepe surfing on the moon"
  image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
  image.save(prompt.replace(" ","_") + ".png")