diff --git a/fooocus_version.py b/fooocus_version.py
index 7a0929f0..778ba2aa 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1,2 +1,2 @@
-version = '1.0.5'
+version = '1.0.6'
 
diff --git a/launch.py b/launch.py
index 51d946d7..67dfad34 100644
--- a/launch.py
+++ b/launch.py
@@ -73,6 +73,9 @@ def download_models():
 
 
 prepare_environment()
+
+import cuda_malloc
+
 download_models()
 
 from webui import *
diff --git a/webui.py b/webui.py
index cb665d57..f2ee76b6 100644
--- a/webui.py
+++ b/webui.py
@@ -1,5 +1,3 @@
-import cuda_malloc
-
 import gradio as gr
 import random
 import fooocus_version