write a comment for fix_checkpoint function

This commit is contained in:
AUTOMATIC 2023-01-19 20:39:03 +03:00
parent 51517f3ea6
commit 6073456c83

View File

@ -69,6 +69,13 @@ def undo_optimizations():
ldm.modules.diffusionmodules.model.AttnBlock.forward = diffusionmodules_model_AttnBlock_forward
def fix_checkpoint():
"""checkpoints are now added and removed in embedding/hypernet code, since torch doesn't want
checkpoints to be added when not training (there's a warning)"""
pass
class StableDiffusionModelHijack:
fixes = None
comments = []