How to use amaai-lab/apex with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="amaai-lab/apex", trust_remote_code=True)
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("amaai-lab/apex", trust_remote_code=True, dtype="auto")
e0ec6d3
1
2
3
4
5
6
from .configuration_apex import APEXConfig from .modeling_apex import APEXModel from transformers import AutoConfig, AutoModel AutoConfig.register("apex", APEXConfig) AutoModel.register(APEXConfig, APEXModel)