littleworth
/

protgpt2-distilled-small

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

littleworth commited on May 7

Commit

9f3bec3

•

1 Parent(s): 99d143c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel, TextGenerationPipeline
 import re
 # Load the model and tokenizer
-model_name = "littleworth/protgpt2-distilled-tiny"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)

 import re
 # Load the model and tokenizer
+model_name = "littleworth/protgpt2-distilled-small"
 tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 model = GPT2LMHeadModel.from_pretrained(model_name)