from transformers import AutoModelWithHeads model = AutoModelWithHeads.from_pretrained("CNXT/CHaTx") model.load_adapter("CNXT/CHaTx", source="hf")