diff --git a/modules/upscaler_utils.py b/modules/upscaler_utils.py
index 1d610dbff..39f78a0b7 100644
--- a/modules/upscaler_utils.py
+++ b/modules/upscaler_utils.py
@@ -6,7 +6,7 @@ import torch
 import tqdm
 from PIL import Image
 
-from modules import images
+from modules import devices, images
 
 logger = logging.getLogger(__name__)