Robo-Diffusion 2 (base)
A dreambooth-method finetune of stable diffusion that will output cool looking robots when prompted.
Usage
Keep the words nousr robot towards the beginning of your prompt to invoke the finetuned style. Use negative prompts to achieve the best result.
import torch
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
scheduler = EulerDiscreteScheduler.from_pretrained("nousr/robo-diffusion-2-base", subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16)
pipe = pipe.to("cuda")
prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."
negative_prompt = "black and white robot, picture frame, a children's drawing in crayon. #Wholesale, Abstract Metal Sculpture. i'm leaving a bad review."
image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=32, guidance_scale=5.0).images[0]
image.save("robo.png")
Original Model
Based on stable diffusion 1.4 can be found here
Socials
Use the #robodiffusion so i can see the cool stuff you make!
If you enjoy the model i'd appreciate a follow on twitter
If you are feeling especially generous, you can sponsor me on github
*NOTE: ensure you have read the license and agree to the terms
- Downloads last month
- 174