Vous avez un workflow ComfyUI ? Obtenez une documentation lisible directement dans votre navigateur.Rendre un workflow lisible.Essayer maintenant ↗
Deepdanbooru-pytorch is a modular, weight-compatible rewrite of TorchDeepDanbooru, with all original weights migrated into a model.safetensors file for efficient loading.
Source du modèle
Description de la source
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 sourceVérifié 17 sept.
Artefacts du modèle
1 artefactExtraits de sources
2 extraits--- 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