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)