fix issues with api_endpoint when training

xieyongliang · xieyongliang · commit 569eb3a8af62 · 2023-03-15T08:47:12.000+08:00
diff --git 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:
