yakine commited on
Commit
49b1c06
·
verified ·
1 Parent(s): 01c0141

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -29,6 +29,7 @@ if not hf_token:
29
  # Load GPT-2 model and tokenizer
30
  tokenizer_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
31
  model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
 
32
 
33
  # Create a pipeline for text generation using GPT-2
34
  text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer_gpt2)
 
29
  # Load GPT-2 model and tokenizer
30
  tokenizer_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
31
  model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
32
+ model_gpt2.config.pad_token_id = model_gpt2.config.eos_token_id
33
 
34
  # Create a pipeline for text generation using GPT-2
35
  text_generator = pipeline("text-generation", model=model_gpt2, tokenizer=tokenizer_gpt2)