Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to get the model? #70

Open
KatiaSishost opened this issue Nov 5, 2022 · 1 comment
Open

How to get the model? #70

KatiaSishost opened this issue Nov 5, 2022 · 1 comment

Comments

@KatiaSishost
Copy link

How to get the model and pass it to use in javascript like tensorflow toxicity model?

@auadams
Copy link

auadams commented Feb 11, 2023

the model downloads when you initialize the package. check your cache you should be able to find it in there. if not check the src you should beable to find the download links in detoxify/detoxify/detoxify.py

DOWNLOAD_URL = "https://github.com/unitaryai/detoxify/releases/download/"
MODEL_URLS = {
    "original": DOWNLOAD_URL + "v0.1-alpha/toxic_original-c1212f89.ckpt",
    "unbiased": DOWNLOAD_URL + "v0.3-alpha/toxic_debiased-c7548aa0.ckpt",
    "multilingual": DOWNLOAD_URL + "v0.4-alpha/multilingual_debiased-0b549669.ckpt",
    "original-small": DOWNLOAD_URL + "v0.1.2/original-albert-0e1d6498.ckpt",
    "unbiased-small": DOWNLOAD_URL + "v0.1.2/unbiased-albert-c8519128.ckpt",
}

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants