andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove dead code. (#13251)

Browse Source
This commit is contained in:
comfyanonymous
2026-04-01 17:22:06 -07:00
committed by GitHub
parent 7d437687c2
commit 0c63b4f6e3
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/modules/encoders/noise_aug_modules.py
+2 -5
View File
@@ -3,12 +3,9 @@ from ..diffusionmodules.openaimodel import Timestep
import torch import torch
class CLIPEmbeddingNoiseAugmentation(ImageConcatWithNoiseAugmentation): class CLIPEmbeddingNoiseAugmentation(ImageConcatWithNoiseAugmentation):
def __init__(self, *args, clip_stats_path=None, timestep_dim=256, **kwargs): def __init__(self, *args, timestep_dim=256, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
if clip_stats_path is None: clip_mean, clip_std = torch.zeros(timestep_dim), torch.ones(timestep_dim)
clip_mean, clip_std = torch.zeros(timestep_dim), torch.ones(timestep_dim)
else:
clip_mean, clip_std = torch.load(clip_stats_path, map_location="cpu")
self.register_buffer("data_mean", clip_mean[None, :], persistent=False) self.register_buffer("data_mean", clip_mean[None, :], persistent=False)
self.register_buffer("data_std", clip_std[None, :], persistent=False) self.register_buffer("data_std", clip_std[None, :], persistent=False)
self.time_embed = Timestep(timestep_dim) self.time_embed = Timestep(timestep_dim)