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