LatentUpscale node needs BatchNorm
Open
Feature
- Dominant language
- Python
- Stars
- 133k
- Forks
- 15.7k
- Avg merge
- 1d 7h
- Merged PRs (30d)
- 158
Description
### Feature Idea
The image, after being encoded by VAE, is transformed into a latent tensor that conforms to the standard Gaussian distribution. However, directly scaling the latent tensor does not follow this rule, which will lead to a lack of practicality in subsequent operations. We hope that the official can add the BatchNorm attribute to the LatentUpscale node
### Existing Solutions
```
norm_layer = nn.BatchNorm2d(feature_dim)
norm_upscale_latent = norm_layer(upscale_sample)
```
### Other
_No response_
Contributor guide
Assessment
This issue has not been assessed yet.