README.md

metadata

language:
  - en
thumbnail: https://huggingface.co/nousr/robo-diffusion/resolve/main/robo_example.png
tags:
  - robots
  - stable-diffusion
  - aiart
  - text-to-image
license: openrail++

Robo-Diffusion 2 (base) - Alpha Release

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(MODEL_ID, subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16, revision="fp16")
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

---

*NOTE: ensure you have read the license and agree to the terms