pepe-diffuser / README.md
Dipl0's picture
Update README.md
9e854d9
|
raw
history blame
549 Bytes
metadata
datasets:
  - Dipl0/pepe_image_stable_diff
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")