nomic-ai

/

nomic-bert-2048

like 52

Follow

Nomic AI 786

Purpose of this change:

• This warning will not trigger when doing AutoModel.from_pretrained('nomic-ai/nomic-embed-text-v1.5') because that is not an MoE model.
• This warning is straightforward to suppress from user code if the user does not want to install Nomic's fork of megablocks:

import warnings
from transformers import AutoModel
with warnings.catch_warnings():
    warnings.filterwarnings('ignore', '.*megablocks.*')
    model = AutoModel.from_pretrained('nomic-ai/nomic-embed-text-v2-moe', trust_remote_code=True)

• The warning mentions Nomic's fork of megablocks so it is clear that pip install megablocks will not help.
• The sad path with the warning will be used when upstream megablocks is detected instead of trying to pass an unrecognized argument and raising an error.