Module: Transformers::XlmRoberta
- Defined in:
- lib/transformers/models/xlm_roberta/modeling_xlm_roberta.rb,
lib/transformers/models/xlm_roberta/configuration_xlm_roberta.rb,
lib/transformers/models/xlm_roberta/tokenization_xlm_roberta_fast.rb
Defined Under Namespace
Classes: XLMRobertaAttention, XLMRobertaClassificationHead, XLMRobertaConfig, XLMRobertaEmbeddings, XLMRobertaEncoder, XLMRobertaForCausalLM, XLMRobertaForMaskedLM, XLMRobertaForMultipleChoice, XLMRobertaForQuestionAnswering, XLMRobertaForSequenceClassification, XLMRobertaForTokenClassification, XLMRobertaIntermediate, XLMRobertaLMHead, XLMRobertaLayer, XLMRobertaModel, XLMRobertaOutput, XLMRobertaPooler, XLMRobertaPreTrainedModel, XLMRobertaSdpaSelfAttention, XLMRobertaSelfAttention, XLMRobertaSelfOutput, XLMRobertaTokenizerFast
Constant Summary collapse
- XLM_ROBERTA_SELF_ATTENTION_CLASSES =
{"eager" => XLMRobertaSelfAttention, "sdpa" => XLMRobertaSdpaSelfAttention}