Commit
•
9e91305
1
Parent(s):
6468fc1
Update README.md
Browse files
README.md
CHANGED
@@ -13,6 +13,20 @@ tags:
|
|
13 |
|
14 |
![So cute, right?](https://huggingface.co/nroggendorff/cats/resolve/main/samples/0024.png)
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
- `train_batch_size`: 16
|
17 |
- `eval_batch_size`: 16
|
18 |
- `num_epochs`: 50
|
@@ -20,4 +34,19 @@ tags:
|
|
20 |
- `learning_rate`: 1e-4
|
21 |
- `lr_warmup_steps`: 500
|
22 |
- `mixed_precision`: "fp16"
|
23 |
-
- `eval_metric`: "mean_squared_error"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
|
14 |
![So cute, right?](https://huggingface.co/nroggendorff/cats/resolve/main/samples/0024.png)
|
15 |
|
16 |
+
### Diffusers
|
17 |
+
```py
|
18 |
+
from diffusers import DiffusionPipeline
|
19 |
+
|
20 |
+
pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
|
21 |
+
pipe = pipe.to("cuda")
|
22 |
+
|
23 |
+
image = pipe().images[0]
|
24 |
+
|
25 |
+
image.save("cat.png")
|
26 |
+
```
|
27 |
+
|
28 |
+
|
29 |
+
### Model Details
|
30 |
- `train_batch_size`: 16
|
31 |
- `eval_batch_size`: 16
|
32 |
- `num_epochs`: 50
|
|
|
34 |
- `learning_rate`: 1e-4
|
35 |
- `lr_warmup_steps`: 500
|
36 |
- `mixed_precision`: "fp16"
|
37 |
+
- `eval_metric`: "mean_squared_error"
|
38 |
+
|
39 |
+
### Bias
|
40 |
+
- This model may exhibit biases due to its training data. It will not display images of abused or sick cats, as it prioritizes the well-being of animals.
|
41 |
+
|
42 |
+
### Limitations
|
43 |
+
|
44 |
+
- The model does not achieve perfect photorealism
|
45 |
+
- The model cannot render legible text
|
46 |
+
- The model was trained on a medium-to-large-scale dataset
|
47 |
+
[Cats vs Dogs](https://www.kaggle.com/datasets/shaunthesheep/microsoft-catsvsdogs-dataset/data)
|
48 |
+
|
49 |
+
### Developed by
|
50 |
+
- Noa Linden Roggendorff
|
51 |
+
|
52 |
+
*This model card was written by Noa Roggendorff and is based on the [Stable Diffusion v1-5 Model Card](https://huggingface.co/runwayml/stable-diffusion-v1-5).*
|