diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-15 11:54:33 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-15 11:54:33 +0000 |
| commit | cdaab233c46d01df45a8890da12ea9840ad6b98d (patch) | |
| tree | 02be6fe4cebf666ace6e29aa64ee8cdd331a33c7 /modules/sd_hijack.py | |
| parent | 6d2eb339d650d3a6d9e4c8f5222d5080b526e7d2 (diff) | |
| parent | 56b0b0feb5b629873983b6e5e5a16c726a802d32 (diff) | |
| download | stable-diffusion-webui-gfx803-cdaab233c46d01df45a8890da12ea9840ad6b98d.tar.gz stable-diffusion-webui-gfx803-cdaab233c46d01df45a8890da12ea9840ad6b98d.tar.bz2 stable-diffusion-webui-gfx803-cdaab233c46d01df45a8890da12ea9840ad6b98d.zip | |
Merge remote-tracking branch 'origin/master'
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index ec7d14cb..65414518 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -57,7 +57,7 @@ def split_cross_attention_forward(self, x, context=None, mask=None): q, k, v = map(lambda t: rearrange(t, 'b n (h d) -> (b h) n d', h=h), (q_in, k_in, v_in))
del q_in, k_in, v_in
- r1 = torch.zeros(q.shape[0], q.shape[1], v.shape[2], device=q.device)
+ r1 = torch.zeros(q.shape[0], q.shape[1], v.shape[2], device=q.device, dtype=q.dtype)
stats = torch.cuda.memory_stats(q.device)
mem_active = stats['active_bytes.all.current']
|