learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-08-14 09:31:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

delete unnecessary code and replace

Browse Source
This commit is contained in:
cjfghk5697
2023-01-19 23:40:23 +09:00
parent 59dde18a94
commit ce816b9be3
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
labml_nn/diffusion/stable_diffusion/model/unet.py
View File

@ -20,7 +20,6 @@ from typing import List
import numpy as np
import torch
import torch as th
import torch.nn as nn
import torch.nn.functional as F
@ -174,7 +173,7 @@ class UNetModel(nn.Module):
x = self.middle_block(x, t_emb, cond)
# Output half of the U-Net
for module in self.output_blocks:
x = th.cat([x, x_input_block.pop()], dim=1)
x = torch.cat([x, x_input_block.pop()], dim=1)
x = module(x, t_emb, cond)
# Final normalization and $3 \times 3$ convolution