TastyPancakes
/
testing_generation
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:EddieCueto/testing_generation

This commit is contained in:
Eduardo Cueto Mendoza 2020-08-24 08:39:21 -06:00
parent e00355bf6e c1cc4f86ac
commit 815c87020e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gen-test.py
View File

@ -8,7 +8,7 @@ gpt2.load_model(force_download=False)
#text = gpt2.generate(return_text=True) # Generates text and returns it in an array
#gpt2.generate(interactive=True, n_samples=1, return_text=True) # A different prompt each time
my_corpus = "./Corpus/Neural_Ordinary_Differential_Equations.txt"
my_corpus = "./Corpus/CORPUS.txt"
custom_text = gpt2.finetune(my_corpus, return_text=False)
#print(custom_text)