You can deploy this model and start earning money today!

pearisli/
deepdanbooru-pytorch

Name: deepdanbooru-pytorch
Availability: InStock

Views

No views yet

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

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)