Update lxmert/src/lxmert_lrp.py

lxmert/src/lxmert_lrp.py

@@ -34,7 +34,8 @@ from transformers.file_utils import (
 )
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import logging
-from transformers.configuration_lxmert import LxmertConfig
+# from transformers.configuration_lxmert import LxmertConfig
+from transformers.models.lxmert.configuration_lxmert import LxmertConfig
 
 logger = logging.get_logger(__name__)
 
@@ -1135,7 +1136,7 @@ class LxmertModel(LxmertPreTrainedModel):
 
     @add_start_docstrings_to_model_forward(LXMERT_INPUTS_DOCSTRING.format("batch_size, sequence_length"))
     @add_code_sample_docstrings(
-        tokenizer_class=_TOKENIZER_FOR_DOC,
+        processor_class=_TOKENIZER_FOR_DOC,
         checkpoint="unc-nlp/lxmert-base-uncased",
         output_type=LxmertModelOutput,
         config_class=_CONFIG_FOR_DOC,
@@ -1624,7 +1625,7 @@ class LxmertForQuestionAnswering(LxmertPreTrainedModel):
 
     @add_start_docstrings_to_model_forward(LXMERT_INPUTS_DOCSTRING.format("batch_size, sequence_length"))
     @add_code_sample_docstrings(
-        tokenizer_class=_TOKENIZER_FOR_DOC,
+        processor_class=_TOKENIZER_FOR_DOC,
         checkpoint="unc-nlp/lxmert-base-uncased",
         output_type=LxmertForQuestionAnsweringOutput,
         config_class=_CONFIG_FOR_DOC,