SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Merge pull request #462 from 8enmann/master

Browse Source 
fix run_gpt2.py
This commit is contained in:
Thomas Wolf
2019-04-11 21:54:46 +02:00
committed by GitHub
parent 2cdfb8b254 fd8a3556f0
commit 616743330e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/run_gpt2.py
View File

@@ -83,7 +83,8 @@ def run_model():
elif args.length > model.config.n_ctx: elif args.length > model.config.n_ctx:
raise ValueError("Can't get samples longer than window size: %s" % model.config.n_ctx) raise ValueError("Can't get samples longer than window size: %s" % model.config.n_ctx)
while not args.unconditional: while True:
context_tokens = []
if not args.unconditional: if not args.unconditional:
raw_text = input("Model prompt >>> ") raw_text = input("Model prompt >>> ")
while not raw_text: while not raw_text:
@@ -106,6 +107,8 @@ def run_model():
print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40) print("=" * 40 + " SAMPLE " + str(generated) + " " + "=" * 40)
print(text) print(text)
print("=" * 80) print("=" * 80)
if args.unconditional:
break
if __name__ == '__main__': if __name__ == '__main__':
run_model() run_model()