Add num_tokens setting

Need this setting too.