diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index 5f91f801a..2b14595f8 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -55,7 +55,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None):
         with devices.autocast(), torch.no_grad():
             x_sample = sd_vae_consistency.decoder_model()(
                 sample.to(devices.device, devices.dtype)/0.18215,
-                schedule=[1.0]
+                schedule=[1.0],
             )
         sd_vae_consistency.unload()
     else:
diff --git a/modules/sd_vae_consistency.py b/modules/sd_vae_consistency.py
index 47e86e6b1..cf189a105 100644
--- a/modules/sd_vae_consistency.py
+++ b/modules/sd_vae_consistency.py
@@ -5,8 +5,6 @@ Improved decoding for stable diffusion vaes.
 https://github.com/openai/consistencydecoder
 """
 import os
-import torch
-import torch.nn as nn
 
 from modules import devices, paths_internal, shared
 from consistencydecoder import ConsistencyDecoder
@@ -32,4 +30,4 @@ def decoder_model():
 def unload():
     global sd_vae_consistency_models
     if sd_vae_consistency_models is not None:
-        sd_vae_consistency_models.ckpt.to('cpu')
\ No newline at end of file
+        sd_vae_consistency_models.ckpt.to('cpu')