Update app.py
Browse files
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(
|
| 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(
|