File size: 895 Bytes
f6142f7 c792bde f6142f7 c792bde eb26634 f6142f7 c792bde 3b7619a c792bde 9817a8f c792bde 9817a8f c792bde |
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 |
---
license: mit
tags:
- deepdanbooru
- safetensors
- pytorch
pipeline_tag: image-to-text
---
# Deepdanbooru-PyTorch Model Card
Deepdanbooru-pytorch is a modular, weight-compatible rewrite of [TorchDeepDanbooru](https://github.com/AUTOMATIC1111/TorchDeepDanbooru), with all original weights migrated into a `model.safetensors` file for efficient loading.
## Model Details
The complete implementation is available in the [Pearisli/VisGen repository](https://github.com/Pearisli/VisGen/blob/main/models/deepdanbooru.py). You don’t need to clone the entire repo—simply copy the contents of `deepdanbooru.py` and you’re all set.
## Examples
```python
# import or define class DeepDanbooruModel
from PIL import Image
model = DeepDanbooruModel.from_pretrained("pearisli/deepdanbooru-pytorch")
model = model.to("cuda")
image = Image.open("example.jpg").convert("RGB")
model.tag(image)
``` |