nousr / robo-diffusion-2-base

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