Hollway
/

gpt2_finetune

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Hollway commited on Jun 21, 2023

Commit

0c6b87b

•

1 Parent(s): 0e34de3

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -12,8 +12,8 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = GPT2Tokenizer.from_pretrained('Hollway/gpt2_finetne')
-model = GPT2LMHeadModel.from_pretrained('Hollway/gpt2_finetne').to(device)
 ```
 ### Inference

 import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+tokenizer = GPT2Tokenizer.from_pretrained('Hollway/gpt2_finetune')
+model = GPT2LMHeadModel.from_pretrained('Hollway/gpt2_finetune').to(device)
 ```
 ### Inference