diff options
| author | unknown <mcgpapu@gmail.com> | 2023-01-28 09:40:51 +0000 |
|---|---|---|
| committer | unknown <mcgpapu@gmail.com> | 2023-01-28 09:40:51 +0000 |
| commit | e79b7db4b47a33889551b9266ee3277879d4f560 (patch) | |
| tree | 1c1944204e58e254bfea22ae44edccdbb54e6b3c /modules/deepbooru_model.py | |
| parent | b921a52071cf2a5e551c31a6073af6eaebbf7847 (diff) | |
| parent | e8a41df49fadd2cf9f23b1f02d75a4947bec5646 (diff) | |
| download | stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.tar.gz stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.tar.bz2 stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.zip | |
Merge branch 'master' of github.com:AUTOMATIC1111/stable-diffusion-webui into gamepad
Diffstat (limited to 'modules/deepbooru_model.py')
| -rw-r--r-- | modules/deepbooru_model.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/deepbooru_model.py b/modules/deepbooru_model.py index edd40c81..83d2ff09 100644 --- a/modules/deepbooru_model.py +++ b/modules/deepbooru_model.py @@ -2,6 +2,8 @@ import torch import torch.nn as nn
import torch.nn.functional as F
+from modules import devices
+
# see https://github.com/AUTOMATIC1111/TorchDeepDanbooru for more
@@ -196,7 +198,7 @@ class DeepDanbooruModel(nn.Module): t_358, = inputs
t_359 = t_358.permute(*[0, 3, 1, 2])
t_359_padded = F.pad(t_359, [2, 3, 2, 3], value=0)
- t_360 = self.n_Conv_0(t_359_padded)
+ t_360 = self.n_Conv_0(t_359_padded.to(self.n_Conv_0.bias.dtype) if devices.unet_needs_upcast else t_359_padded)
t_361 = F.relu(t_360)
t_361 = F.pad(t_361, [0, 1, 0, 1], value=float('-inf'))
t_362 = self.n_MaxPool_0(t_361)
|