Commit 23524ad8 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove print.

parent c02baed0
...@@ -366,7 +366,6 @@ def encode_adm(noise_augmentor, conds, batch_size, device): ...@@ -366,7 +366,6 @@ def encode_adm(noise_augmentor, conds, batch_size, device):
#TODO: add a way to control this #TODO: add a way to control this
noise_augment = 0.05 noise_augment = 0.05
noise_level = round((noise_augmentor.max_noise_level - 1) * noise_augment) noise_level = round((noise_augmentor.max_noise_level - 1) * noise_augment)
print(noise_level)
c_adm, noise_level_emb = noise_augmentor(adm_out[:, :noise_augmentor.time_embed.dim], noise_level=torch.tensor([noise_level], device=device)) c_adm, noise_level_emb = noise_augmentor(adm_out[:, :noise_augmentor.time_embed.dim], noise_level=torch.tensor([noise_level], device=device))
adm_out = torch.cat((c_adm, noise_level_emb), 1) adm_out = torch.cat((c_adm, noise_level_emb), 1)
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment