chat/base.py: extend checkpoint_dir before accessing it

[Andrei-Aksionov]

Hi there 👋

Inside auto_download_checkpoint the code also extends the path by prepending checkpoints/ if it's not provided.

Since currently it's done after check_file_size_on_cpu_and_warn, the command

litgpt chat google/gemma-2-9b-it

will fail.