Vector Quantized Models for Planning