Comfy-Org / Comfy-Org/ComfyUI

LatentUpscale node needs BatchNorm

Open
#4,515 0 comments 0 reactions 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.