metadata
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, with all original weights migrated into a model.safetensors file for efficient loading.
Model Details
The model implementation is available in the deepdanbooru.py.
Examples
# 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)