Tem um workflow do ComfyUI? Gere uma documentação legível direto no navegador.Torne um workflow legível.Experimentar agora ↗
Deepdanbooru-pytorch is a modular, weight-compatible rewrite of TorchDeepDanbooru, with all original weights migrated into a model.safetensors file for efficient loading.
Fonte do modelo
Descrição da fonte
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.
Fontes
1 fonteVerificado 17 de set.
Artefatos de modelo
1 artefatoTrechos de fonte
2 trechos--- 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