diff --git a/modules/core.py b/modules/core.py
index 0d3a2c4f..39d3cb8e 100644
--- a/modules/core.py
+++ b/modules/core.py
@@ -1,3 +1,5 @@
+import cuda_malloc
+
 import os
 import random
 import cv2
diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py
index d81aea03..84e18b24 100644
--- a/modules/default_pipeline.py
+++ b/modules/default_pipeline.py
@@ -1,6 +1,6 @@
+import modules.core as core
 import os
 import torch
-import modules.core as core
 
 from modules.path import modelfile_path