fix issues with api_endpoint when training

AIRI-AUBE · Mar 15, 2023 · 569eb3a · 569eb3a
1 parent 460fe3b
commit 569eb3a
Showing 1 changed file with 2 additions and 0 deletions.
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: