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}