query_model "Name three TPU generations." max_tokens=64