MythSus commited on
Commit
26186cb
·
verified ·
1 Parent(s): 456b259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from peft import PeftModel
7
 
8
 
9
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
10
- tokenizer.pad_token = self.tokenizer.eos_token
11
  base_model = GPT2LMHeadModel.from_pretrained("gpt2-medium")
12
  finetuned_model = PeftModel.from_pretrained(base_model, "./")
13
  finetuned_model.eval()
 
7
 
8
 
9
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2-medium")
10
+ tokenizer.pad_token = tokenizer.eos_token
11
  base_model = GPT2LMHeadModel.from_pretrained("gpt2-medium")
12
  finetuned_model = PeftModel.from_pretrained(base_model, "./")
13
  finetuned_model.eval()