aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-02-04 12:23:16 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-02-04 12:23:16 +0000
commit3e0f9a75438fa815429b5530261bcf7d80f3f101 (patch)
tree75cdd79444f2950583de3eb1609a1fd5648d9f25 /modules/sd_models.py
parent40e51fd6efa9c09a82c5ab391dbbd2c806971582 (diff)
downloadstable-diffusion-webui-gfx803-3e0f9a75438fa815429b5530261bcf7d80f3f101.tar.gz
stable-diffusion-webui-gfx803-3e0f9a75438fa815429b5530261bcf7d80f3f101.tar.bz2
stable-diffusion-webui-gfx803-3e0f9a75438fa815429b5530261bcf7d80f3f101.zip
fix issue with switching back to checkpoint that had its checksum calculated during runtime mentioned in #7506
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r--modules/sd_models.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py
index 0e61d323..af1731e5 100644
--- a/modules/sd_models.py
+++ b/modules/sd_models.py
@@ -65,10 +65,11 @@ class CheckpointInfo:
self.shorthash = self.sha256[0:10]
if self.shorthash not in self.ids:
- self.ids += [self.shorthash, self.sha256]
- self.register()
+ self.ids += [self.shorthash, self.sha256, f'{self.name} [{self.shorthash}]']
+ checkpoints_list.pop(self.title)
self.title = f'{self.name} [{self.shorthash}]'
+ self.register()
return self.shorthash