How to use mohammadmahdinouri/mol-base-init-pruned with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="mohammadmahdinouri/mol-base-init-pruned")
# Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("mohammadmahdinouri/mol-base-init-pruned", dtype="auto")
