How to use PantagrueLLM/text-base-wiki-mlm with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="PantagrueLLM/text-base-wiki-mlm", trust_remote_code=True)
# Load model directly from transformers import AutoModelForMaskedLM model = AutoModelForMaskedLM.from_pretrained("PantagrueLLM/text-base-wiki-mlm", trust_remote_code=True, device_map="auto")
How to use PantagrueLLM/text-base-wiki-mlm with Fairseq:
from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub models, cfg, task = load_model_ensemble_and_task_from_hf_hub( "PantagrueLLM/text-base-wiki-mlm" )
