pipe.vae = vae                                                                                                                                                                                                                                