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