Commit 6073456c authored by AUTOMATIC's avatar AUTOMATIC
Browse files

write a comment for fix_checkpoint function

parent 51517f3e
Loading
Loading
Loading
Loading
+7 −0
Original line number Original line Diff line number Diff line
@@ -69,6 +69,13 @@ def undo_optimizations():
    ldm.modules.diffusionmodules.model.AttnBlock.forward = diffusionmodules_model_AttnBlock_forward
    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:
class StableDiffusionModelHijack:
    fixes = None
    fixes = None
    comments = []
    comments = []