|
|
--- |
|
|
tags: |
|
|
- stable-diffusion |
|
|
- text-to-image |
|
|
- image-to-image |
|
|
- disney |
|
|
license: mit |
|
|
--- |
|
|
|
|
|
# Disney-Style Portrait Generator |
|
|
|
|
|
This model is fine-tuned on Disney-style images using Stable Diffusion. It generates Disney-style portraits from text prompts. |
|
|
|
|
|
## Usage |
|
|
|
|
|
```python |
|
|
from diffusers import StableDiffusionPipeline |
|
|
import torch |
|
|
|
|
|
# Load the model |
|
|
model = StableDiffusionPipeline.from_pretrained("Molkaatb/DisneyModel", torch_dtype=torch.float16).to("cuda") |
|
|
|
|
|
# Generate an image |
|
|
prompt = "A Disney-style portrait of a princess" |
|
|
image = model(prompt).images[0] |
|
|
image.save("disney_princess.png") |
