mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-11-21 03:11:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

as per wfjsw's suggestion, revert changes for sd_hijack_checkpoint.py

Browse Source
This commit is contained in:
AUTOMATIC1111 2024-06-08 10:54:41 +03:00
parent ad229fae43
commit 603509ec90
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/sd_hijack_checkpoint.py
View File

@ -4,19 +4,16 @@ import ldm.modules.attention
import ldm.modules.diffusionmodules.openaimodel
# Setting flag=False so that torch skips checking parameters.
# parameters checking is expensive in frequent operations.
def BasicTransformerBlock_forward(self, x, context=None):
return checkpoint(self._forward, x, context, flag=False)
return checkpoint(self._forward, x, context)
def AttentionBlock_forward(self, x):
return checkpoint(self._forward, x, flag=False)
return checkpoint(self._forward, x)
def ResBlock_forward(self, x, emb):
return checkpoint(self._forward, x, emb, flag=False)
return checkpoint(self._forward, x, emb)
stored = []