archit11 commited on
Commit
11f5c2d
·
verified ·
1 Parent(s): c5d64f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ max_length = 8192
11
  model_name = "archit11/gpt2-long-finetuned"
12
  config = GPT2Config.from_pretrained(model_name)
13
  model = GPT2LMHeadModel.from_pretrained(model_name)
14
- tokenizer = GPT2Tokenizer.from_pretrained(model_name)
15
 
16
  # Add rotary embeddings
17
  rotary_emb = RotaryEmbedding(
 
11
  model_name = "archit11/gpt2-long-finetuned"
12
  config = GPT2Config.from_pretrained(model_name)
13
  model = GPT2LMHeadModel.from_pretrained(model_name)
14
+ tokenizer = GPT2Tokenizer.from_pretrained("openai-community/gpt2")
15
 
16
  # Add rotary embeddings
17
  rotary_emb = RotaryEmbedding(