skytnt
/

gpt2-japanese-lyric-small

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

skytnt commited on Jul 6, 2022

Commit

72268b6

·

1 Parent(s): e47df31

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -31,7 +31,7 @@ model = GPT2LMHeadModel.from_pretrained("skytnt/gpt2-japanese-lyric-small")
 def gen_lyric(prompt_text: str):
-    prompt_text = "<s>" + prompt_text
     prompt_tokens = tokenizer.tokenize(prompt_text)
     prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
     prompt_tensor = torch.LongTensor(prompt_token_ids).to(device)

 def gen_lyric(prompt_text: str):
+    prompt_text = "<s>" + prompt_text.replace("\n", "\\n ")
     prompt_tokens = tokenizer.tokenize(prompt_text)
     prompt_token_ids = tokenizer.convert_tokens_to_ids(prompt_tokens)
     prompt_tensor = torch.LongTensor(prompt_token_ids).to(device)