From 569eb3a8af626eae16c52d298f74ea118b407b73 Mon Sep 17 00:00:00 2001
From: xieyongliang <yongliangxie@hotmail.com>
Date: Wed, 15 Mar 2023 08:47:12 +0800
Subject: [PATCH] fix issues with api_endpoint when training

---
 modules/shared.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/shared.py b/modules/shared.py
index 883fc62afc4..02cd12139bb 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -145,6 +145,8 @@
     sd_model_checkpoint_component = None
     create_train_dreambooth_component = None
     username = ''
+else:
+    api_endpoint = cmd_opts.api_endpoint
 
 response = requests.get(url=f'{api_endpoint}/sd/industrialmodel')
 if response.status_code == 200: