Fix loading more recent VaeKl checkpoints

[jonatanklosko]

https://github.com/huggingface/diffusers/pull/3387 changed layer names in the VaeKl implementation, so in order to load newer and older checkpoints we need to check for both layer names. In particular this is relevant to fp16 checkpoints for SDv1-4.

I changed the params mapping to support multiple layer names to try like this:

%{
  ...,
  "attentions.{m}.query" => ["attentions.{m}.to_q", "attentions.{m}.query"],
  ...
}