How to use AlbertHSU/BertTEST with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("AlbertHSU/BertTEST", device_map="auto")