From 14390356dd0c29c5611d480fa4ad8735e6dff447 Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Thu, 10 Aug 2023 13:52:53 -0700
Subject: [PATCH] i

---
 modules/core.py             | 2 ++
 modules/default_pipeline.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

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