VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 23:01:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not convert to float

Browse Source
This commit is contained in:
rockerBOO
2025-04-15 04:41:59 -04:00
parent 4d005cdf3d
commit c2f75f43a4
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
library/model_utils.py
View File

@@ -9,10 +9,10 @@ class AID(nn.Module):
def forward(self, x: Tensor):
if self.training:
pos_mask = (x >= 0).float() * torch.bernoulli(torch.ones_like(x) * self.p)
neg_mask = (x < 0).float() * torch.bernoulli(torch.ones_like(x) * (1 - self.p))
pos_mask = (x >= 0) * torch.bernoulli(torch.ones_like(x) * self.p)
neg_mask = (x < 0) * torch.bernoulli(torch.ones_like(x) * (1 - self.p))
return x * (pos_mask + neg_mask)
else:
pos_part = (x >= 0).float() * x * self.p
neg_part = (x < 0).float() * x * (1 - self.p)
pos_part = (x >= 0) * x * self.p
neg_part = (x < 0) * x * (1 - self.p)
return pos_part + neg_part