diff options
| author | aria1th <35677394+aria1th@users.noreply.github.com> | 2022-11-04 07:04:03 +0000 |
|---|---|---|
| committer | aria1th <35677394+aria1th@users.noreply.github.com> | 2022-11-04 07:04:03 +0000 |
| commit | f5d394214d6ee74a682d0a1016bcbebc4b43c13a (patch) | |
| tree | c7d5d59bf772b5dfe551b65221a56d5bdb81b4cb | |
| parent | 283249d2390f0f3a1c8a55d5d9aa551e3e9b2f9c (diff) | |
| download | stable-diffusion-webui-gfx803-f5d394214d6ee74a682d0a1016bcbebc4b43c13a.tar.gz stable-diffusion-webui-gfx803-f5d394214d6ee74a682d0a1016bcbebc4b43c13a.tar.bz2 stable-diffusion-webui-gfx803-f5d394214d6ee74a682d0a1016bcbebc4b43c13a.zip | |
split before declaring file name
| -rw-r--r-- | modules/hypernetworks/hypernetwork.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index de8688a9..9b6a3e62 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -382,6 +382,7 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log shared.state.textinfo = "Initializing hypernetwork training..."
shared.state.job_count = steps
+ hypernetwork_name = hypernetwork_name.rsplit('(', 1)[0]
filename = os.path.join(shared.cmd_opts.hypernetwork_dir, f'{hypernetwork_name}.pt')
log_directory = os.path.join(log_directory, datetime.datetime.now().strftime("%Y-%m-%d"), hypernetwork_name)
@@ -393,7 +394,6 @@ def train_hypernetwork(hypernetwork_name, learn_rate, batch_size, data_root, log else:
hypernetwork_dir = None
- hypernetwork_name = hypernetwork_name.rsplit('(', 1)[0]
if create_image_every > 0:
images_dir = os.path.join(log_directory, "images")
os.makedirs(images_dir, exist_ok=True)
|