diff options
| author | InvincibleDude <81354513+InvincibleDude@users.noreply.github.com> | 2023-01-29 11:36:10 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-29 11:36:10 +0000 |
| commit | ee3d63b6beb88e63542976a1095d4c8aa97388bd (patch) | |
| tree | cf891130682c343107ae0a0f8cec309aea16807a /modules/deepbooru_model.py | |
| parent | 44c0e6b993d00bb2f441f0fde409bcb79136f034 (diff) | |
| parent | 00dab8f10defbbda579a1bc89c8d4e972c58a20d (diff) | |
| download | stable-diffusion-webui-gfx803-ee3d63b6beb88e63542976a1095d4c8aa97388bd.tar.gz stable-diffusion-webui-gfx803-ee3d63b6beb88e63542976a1095d4c8aa97388bd.tar.bz2 stable-diffusion-webui-gfx803-ee3d63b6beb88e63542976a1095d4c8aa97388bd.zip | |
Merge branch 'master' into master
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)
|