Deepdanbooru Pytorch

--- 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 model implementation is available in the [deepdanbooru.py](https://huggingface.co/pearisli/deepdanbooru-pytorch/blob/main/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) ```