added revert

This commit is contained in:
Nick Walton
2019-11-25 18:13:06 -07:00
parent 5e910bb6c8
commit e2703e535c
+1 -1
View File
@@ -10,7 +10,7 @@ import numpy as np
class GPT2Generator:
def __init__(self, generate_num=60, temperature=0.4, top_k=40, top_p=0.9):
def __init__(self, generate_num=75, temperature=0.4, top_k=40, top_p=0.9):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k