Skip to main content

Use a finetune of Llama-7b and llama cpp to predict Danbooru tags from natural text.

Project description

text2tags

Text2tags is a Python library that uses a finetune of Llama-7b and llama cpp to predict Danbooru tags from natural text.

Installation

Use the package manager [pip]

pip install text2tags-lib

Usage

from text2tags import TaggerLlama

model = TaggerLlama()

tags = model.predict_tags("Minato Aqua from hololive with pink and blue twintails in a blue maid outfit")

print(', '.join(tags))

Demos

Open In Colab Spaces

Other links

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

text2tags-lib-0.0.7.tar.gz (75.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

text2tags_lib-0.0.7-py3-none-any.whl (74.3 kB view details)

Uploaded Python 3

File details

Details for the file text2tags-lib-0.0.7.tar.gz.

File metadata

  • Download URL: text2tags-lib-0.0.7.tar.gz
  • Upload date:
  • Size: 75.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.11

File hashes

Hashes for text2tags-lib-0.0.7.tar.gz
Algorithm Hash digest
SHA256 0b676dae26c0cd668638c0dfa162f062e1c5504ef70260cbe61674866c0c8c69
MD5 5154437958e86ff19c95fc6fc2868090
BLAKE2b-256 9fc8c4bba5b4bcc672c1095d9ad966ca2491e4b0fac2b97cfba513ed986f9e1e

See more details on using hashes here.

File details

Details for the file text2tags_lib-0.0.7-py3-none-any.whl.

File metadata

  • Download URL: text2tags_lib-0.0.7-py3-none-any.whl
  • Upload date:
  • Size: 74.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.11

File hashes

Hashes for text2tags_lib-0.0.7-py3-none-any.whl
Algorithm Hash digest
SHA256 9a4e32f5cb1fa5cce5f1fadb315f01eaa30fdcec7479723c7bcc304ed5837455
MD5 082d90f6e8608d9bd870eeb216c11b44
BLAKE2b-256 e16834cb65c52774a20f40fc95f93d29a48c1de611f3fc18bc42cd1158b4d74d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page