Have a ComfyUI workflow? Get a readable Workflow Doc in your browser.Make a workflow readable.Try it now ↗
Deepdanbooru-pytorch is a modular, weight-compatible rewrite of TorchDeepDanbooru, with all original weights migrated into a model.safetensors file for efficient loading.
Model source
Source description
Deepdanbooru-pytorch is a modular, weight-compatible rewrite of TorchDeepDanbooru, with all original weights migrated into a model.safetensors file for efficient loading.
The model implementation is available in the deepdanbooru.py.
Sources
1 sourceVerified Sep 17
Model artifacts
1 artifactSource excerpts
2 excerpts--- 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) ```
# 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)
Source context: 51 downloads · 0 likes · Pipeline image-to-text · Repo pearisli/deepdanbooru-pytorch