a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

Fix greyscale injector

Browse Source
This commit is contained in:
James Betker 2020-09-02 10:29:40 -06:00
parent 8b52d46847
commit d90c96e55e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/steps/injectors.py
View File

@ -101,5 +101,5 @@ class GreyInjector(Injector):
def forward(self, state): def forward(self, state):
mean = torch.mean(state[self.opt['in']], dim=1, keepdim=True) mean = torch.mean(state[self.opt['in']], dim=1, keepdim=True)
mean = torch.repeat(mean, (-1, 3, -1, -1)) mean = mean.repeat((1, 3, 1, 1))
return {self.opt['out']: mean} return {self.opt['out']: mean}