README.md

metadata

license: mit
tags:
  - deepdanbooru
  - image-tagging
  - safetensors
  - pytorch

Deepdanbooru-PyTorch Model Card

Deepdanbooru-pytorch is a modular, weight-compatible rewrite of AUTOMATIC1111/TorchDeepDanbooru, with all original weights migrated into a model.safetensors file for efficient loading.

Model Details

The full implementation can be found in Pearisli/VisGen.

Examples

from models import 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)