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")
The community tab is the place to discuss and collaborate with the HF community!