Downloads · 30 days
4
2% of all-time downloads
Elise-hf/distilbert-base-uncased_reddit_categories_unipelft
distilbert-base-uncased_reddit_categories_unipelft is a machine learning model from Elise-hf. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for adapter-transformers.
An adapter for the distilbert-base-uncased model that was trained on the redditcategoriesclean dataset and includes a prediction head for classification.
Downloads · 30 days
4
2% of all-time downloads
All-time downloads
178
Public
Repo size
51 MB
Likes
0
Public
Click a slice to open those files.
.bin25.5 MB · 100%
From the Hugging Face model README
Elise-hf/distilbert-base-uncased_reddit_categories_unipelft for distilbert-base-uncasedAn adapter for the distilbert-base-uncased model that was trained on the reddit_categories_clean dataset and includes a prediction head for classification.
This adapter was created for usage with the adapter-transformers library.
First, install adapter-transformers:
pip install -U adapter-transformers
Note: adapter-transformers is a fork of transformers that acts as a drop-in replacement with adapter support. More
Now, the adapter can be loaded and activated like this:
from transformers import AutoAdapterModel
model = AutoAdapterModel.from_pretrained("distilbert-base-uncased")
adapter_name = model.load_adapter("Elise-hf/distilbert-base-uncased_reddit_categories_unipelft", source="hf", set_active=True)