wang-sy
/

stinfo-gpt2

Text Generation

Inference Endpoints

Model card Files Files and versions Community

wang-sy commited on Mar 3, 2023

Commit

cdc37bd

·

1 Parent(s): da763fd

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ seed = 1337
 device = 'cpu'  # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
 dtype = 'float16'
-output_dir = "."
 model = GPT.from_pretrained(output_dir)
 tokenizer = GPT2Tokenizer.from_pretrained(output_dir)

 device = 'cpu'  # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
 dtype = 'float16'
+output_dir = ""
 model = GPT.from_pretrained(output_dir)
 tokenizer = GPT2Tokenizer.from_pretrained(output_dir)