Skip to content

Commit 1ca0bcd

Browse files
committed
only save if option is enabled
1 parent f5d3942 commit 1ca0bcd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/hypernetworks/hypernetwork.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def save(self, filename):
188188
optimizer_saved_dict['optimizer_name'] = self.optimizer_name
189189

190190
torch.save(state_dict, filename)
191-
if self.optimizer_state_dict:
191+
if shared.opts.save_optimizer_state and self.optimizer_state_dict:
192192
optimizer_saved_dict['hash'] = sd_models.model_hash(filename)
193193
optimizer_saved_dict['optimizer_state_dict'] = self.optimizer_state_dict
194194
torch.save(optimizer_saved_dict, filename + '.optim')

0 commit comments

Comments
 (0)