pepe-diffuser / README.md
Dipl0's picture
Create README.md
cad2481
|
raw
history blame
435 Bytes

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")