README.md

---
datasets:
- huggan/wikiart
base_model:
- CompVis/stable-diffusion-v1-4
---
# Model Card: Stable Diffusion v1-4 Fine-tuned on WikiArt

<!-- Provide a quick summary of what the model is/does. -->
This model is a fine-tuned version of the **Stable Diffusion v1.4** model on the **WikiArt** dataset.  
It can generate images in various artistic styles learned from WikiArt.

## Model Details

### Model Description

- **Developed by:** potatopizza
- **Funded by [optional]:** N/A
- **Shared by [optional]:** potatopizza
- **Model type:** Text-to-Image
- **Language(s) (NLP):** N/A (this is a text-to-image generation model)
- **License:** [CreativeML Open RAIL-M](https://huggingface.co/spaces/CompVis/stable-diffusion-license) 
- **Finetuned from model:** [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4)

### Model Sources

- **Repository (Hugging Face):** [YourUsername/Stable-Diffusion-v1-4-wikiart](https://huggingface.co/YourUsername/Stable-Diffusion-v1-4-wikiart)

## Uses

### Direct Use
- Text-to-Image generation, specifically in various WikiArt styles.

### Out-of-Scope Use
- Generating harmful, explicit, or disallowed content per the license.
- Any usage that violates the **CreativeML Open RAIL** license terms.

## Bias, Risks, and Limitations
- The generated images might reflect biases present in the WikiArt dataset (art style bias, cultural bias, etc.).
- Potential misuse for generating disallowed content (users should comply with the usage policy).

## How to Get Started with the Model

Use Model code:

```python
from diffusers import StableDiffusionPipeline
import torch

model_id = "YourUsername/Stable-Diffusion-v1-4-wikiart"  # 변경 필요
pipe = StableDiffusionPipeline.from_pretrained(
    model_id,
    torch_dtype=torch.float16
).to("cuda")

prompt = "A beautiful painting of a sunset over the mountains, in the style of eric fischl"
image = pipe(prompt).images[0]
image.save("example.png")