From e1d0b738507b2da082519a27b5953d0d754777bc Mon Sep 17 00:00:00 2001
From: lvmin <lyuminzhang@outlook.com>
Date: Fri, 11 Aug 2023 09:55:14 -0700
Subject: [PATCH] i

---
 fooocus_version.py | 2 +-
 launch.py          | 3 +++
 webui.py           | 2 --
 3 files changed, 4 insertions(+), 3 deletions(-)

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