Update lxmert/src/huggingface_lxmert.py

lxmert/src/huggingface_lxmert.py

@@ -35,7 +35,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__)
@@ -930,7 +931,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,
@@ -1411,7 +1412,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,