diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py
index 8bbd8caba..dd9e10a8b 100644
--- a/modules/sd_samplers_common.py
+++ b/modules/sd_samplers_common.py
@@ -1,5 +1,6 @@
 import inspect
 from collections import namedtuple
+from contextlib import nullcontext
 import torch
 from PIL import Image
 from modules import devices, images, sd_vae_approx, sd_samplers, sd_vae_taesd, shared, sd_models