Hi, thanks for your great works! I found the `inference_wrapper` does not performs correctly: The outputed results seem odd.  chroma was downloaded from hf. I tried both v25 and v26 and got same results. VAE and t5 downloaded from the link you provided, all keys matched.