mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-21 07:30:02 +08:00
10 lines
313 B
Python
10 lines
313 B
Python
|
from torch.utils.checkpoint import checkpoint
|
||
|
|
|
||
|
|
def BasicTransformerBlock_forward(self, x, context=None):
|
||
|
|
return checkpoint(self._forward, x, context)
|
||
|
|
|
||
|
|
def AttentionBlock_forward(self, x):
|
||
|
|
return checkpoint(self._forward, x)
|
||
|
|
|
||
|
|
def ResBlock_forward(self, x, emb):
|
||
|
|
return checkpoint(self._forward, x, emb)
|