File size: 1,515 Bytes
ee2f640 3a1fba5 ee2f640 3a1fba5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
---
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.
<img src="https://huggingface.co/nousr/robo-diffusion-2-base/resolve/main/example_grid.png"/>
# 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.
```python
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](https://huggingface.co/nousr/robo-diffusion)
---
*NOTE: ensure you have read the license and agree to the terms |