Issues with prefill & generate

[qihqi]

As reported by @tengomucho

Currently there are a few issues with prefill / generate implemention:

1. Prefill does not use self._sample to do sampling.
2. Prefill returns a token, so first time generate calls it should return the second generated token, but now it returns the first token again. This is historical but quite unintuitive.