Skip to content

Commit 280aa3c

Browse files
committed
change default parameter option
1 parent c2c5e23 commit 280aa3c

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

main.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,12 @@ def text_generator(state_dict):
2323
parser.add_argument('--unconditional', action='store_true', help='If true, unconditional generation.')
2424
parser.add_argument("--batch_size", type=int, default=-1)
2525
parser.add_argument("--length", type=int, default=-1)
26-
parser.add_argument("--temperature", type=int, default=1)
27-
parser.add_argument("--top_k", type=int, default=0)
26+
parser.add_argument("--temperature", type=int, default=0.7)
27+
parser.add_argument("--top_k", type=int, default=40)
2828
args = parser.parse_args()
29-
print(args)
29+
30+
if args.quiet is False:
31+
print(args)
3032

3133
if args.batch_size == -1:
3234
args.batch_size = 1

0 commit comments

Comments
 (0)