How to use Alexei1/imdb with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Alexei1/imdb", dtype="auto")