Andrew Ryzhkov

Results 31 comments of Andrew Ryzhkov

Sure: ``` @torch.no_grad() def convert_vae_encoder_onnx(pipe:StableDiffusionPipeline, onnx_path:Path): """ Convert VAE model to ONNX, then IR format. Function accepts pipeline, creates wrapper class for export only necessary for inference part, prepares example...