Skip to content

Commit 9ec8be5

Browse files
TransfoXL is deprecated, don't keep it in tested examples! (#37707)
* TransfoXL is deprecated, so we should remove it from examples that get tested * Remove the tokenizer too * Trigger tests
1 parent be9b0e8 commit 9ec8be5

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

examples/pytorch/text-generation/run_generation.py

+1-7
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,6 @@
3838
OpenAIGPTLMHeadModel,
3939
OpenAIGPTTokenizer,
4040
OPTForCausalLM,
41-
TransfoXLLMHeadModel,
42-
TransfoXLTokenizer,
4341
XLMTokenizer,
4442
XLMWithLMHeadModel,
4543
XLNetLMHeadModel,
@@ -62,7 +60,6 @@
6260
"ctrl": (CTRLLMHeadModel, CTRLTokenizer),
6361
"openai-gpt": (OpenAIGPTLMHeadModel, OpenAIGPTTokenizer),
6462
"xlnet": (XLNetLMHeadModel, XLNetTokenizer),
65-
"transfo-xl": (TransfoXLLMHeadModel, TransfoXLTokenizer),
6663
"xlm": (XLMWithLMHeadModel, XLMTokenizer),
6764
"gptj": (GPTJForCausalLM, AutoTokenizer),
6865
"bloom": (BloomForCausalLM, BloomTokenizerFast),
@@ -368,10 +365,7 @@ def main():
368365
prepare_input = PREPROCESSING_FUNCTIONS.get(args.model_type)
369366
preprocessed_prompt_text = prepare_input(args, model, tokenizer, prompt_text)
370367

371-
if model.__class__.__name__ in ["TransfoXLLMHeadModel"]:
372-
tokenizer_kwargs = {"add_space_before_punct_symbol": True}
373-
else:
374-
tokenizer_kwargs = {}
368+
tokenizer_kwargs = {}
375369

376370
encoded_prompt = tokenizer.encode(
377371
preprocessed_prompt_text, add_special_tokens=False, return_tensors="pt", **tokenizer_kwargs

0 commit comments

Comments
 (0)