mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2026-02-14 21:04:19 +01:00
Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2d67d5821e |
@@ -580,7 +580,7 @@ class ChanLayerNorm(nn.Module):
|
|||||||
|
|
||||||
var = torch.var(x, dim = 1, unbiased = False, keepdim = True)
|
var = torch.var(x, dim = 1, unbiased = False, keepdim = True)
|
||||||
mean = torch.mean(x, dim = 1, keepdim = True)
|
mean = torch.mean(x, dim = 1, keepdim = True)
|
||||||
return (x - mean) * (var + self.eps).rsqrt() * self.g
|
return (x - mean) * (var + eps).rsqrt() * self.g
|
||||||
|
|
||||||
class Residual(nn.Module):
|
class Residual(nn.Module):
|
||||||
def __init__(self, fn):
|
def __init__(self, fn):
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
__version__ = '1.4.1'
|
__version__ = '1.4.2'
|
||||||
|
|||||||
Reference in New Issue
Block a user