How to use mohammadmahdinouri/mol-base-init-pruned-final with Transformers:
# Load model directly from transformers import DistillationWrapper model = DistillationWrapper.from_pretrained("mohammadmahdinouri/mol-base-init-pruned-final", dtype="auto")
