File size: 2,007 Bytes
1d1a873
 
 
 
 
 
66a3420
1d1a873
 
66a3420
 
1d1a873
 
 
 
 
5ad8a8c
66a3420
5ad8a8c
66a3420
5ad8a8c
66a3420
1d1a873
66a3420
1d1a873
e4b8bab
1d1a873
 
 
 
66a3420
1d1a873
 
66a3420
 
1d1a873
 
66a3420
e4b8bab
1d1a873
fc585e8
 
 
 
 
b6a0d96
fc585e8
1d1a873
 
5ad8a8c
1d1a873
66a3420
 
 
1d1a873
fc585e8
66a3420
 
 
 
1d1a873
5ad8a8c
66a3420
 
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
---
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
- **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):** [potatopizza/Stable-Diffusion-v1-4-wikiart](https://huggingface.co/potatopizza/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 

## Example Inference Result

Below is an example image generated using the prompt:
**A painting of the Eiffel Tower in the style of Eric Fischl.**

![Generated Image](image.png)

## How to Get Started with the Model

Use Model code:

```python
from diffusers import StableDiffusionPipeline
import torch

model_id = "potatopizza/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")