diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-04 14:40:19 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-04 14:40:19 +0000 |
| commit | da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6 (patch) | |
| tree | a2eec9c47e820e7ab351337f73c99d874b4b904f /modules/sd_hijack_checkpoint.py | |
| parent | cffc240a7327ae60671ff533469fc4ed4bf605de (diff) | |
| parent | 47df0849019abac6722c49512f4dd2285bff5b7d (diff) | |
| download | stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.tar.gz stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.tar.bz2 stable-diffusion-webui-gfx803-da5c1e8a732c173ed8ccda9fa32f9a194ff91ab6.zip | |
Merge branch 'master' into inpaint_textual_inversion
Diffstat (limited to 'modules/sd_hijack_checkpoint.py')
| -rw-r--r-- | modules/sd_hijack_checkpoint.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/modules/sd_hijack_checkpoint.py b/modules/sd_hijack_checkpoint.py new file mode 100644 index 00000000..5712972f --- /dev/null +++ b/modules/sd_hijack_checkpoint.py @@ -0,0 +1,10 @@ +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)
\ No newline at end of file |