diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py
index 8fd1ecc3..f357da5a 100644
--- a/api/onnx_web/convert/diffusion/diffusers.py
+++ b/api/onnx_web/convert/diffusion/diffusers.py
@@ -216,7 +216,6 @@ def convert_diffusion_diffusers(
         # SINGLE VAE
         vae_only = pipeline.vae
         vae_latent_channels = vae_only.config.latent_channels
-        vae_out_channels = vae_only.config.out_channels
         # forward only through the decoder part
         vae_only.forward = vae_only.decode
         onnx_export(
@@ -266,7 +265,6 @@ def convert_diffusion_diffusers(
         # VAE DECODER
         vae_decoder = pipeline.vae
         vae_latent_channels = vae_decoder.config.latent_channels
-        vae_out_channels = vae_decoder.config.out_channels
         # forward only through the decoder part
         vae_decoder.forward = vae_encoder.decode
         onnx_export(
diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py
index 44addbe6..e3b54510 100644
--- a/api/onnx_web/diffusion/load.py
+++ b/api/onnx_web/diffusion/load.py
@@ -3,7 +3,6 @@ from os import path
 from typing import Any, Optional, Tuple
 
 import numpy as np
-import torch
 from diffusers import (
     DDIMScheduler,
     DDPMScheduler,