Add num_tokens setting
Merge pull request #3 from virtuald/multinode

Add num_tokens setting