Remove comment.

This commit is contained in:
comfyanonymous 2025-06-14 18:28:46 -07:00 committed by GitHub
parent aa4f3bad8c
commit 2b2f98c822
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1035,8 +1035,6 @@ class CosmosPredict2(BaseModel):
if sigma_noise_augmentation != 0:
latent_image = latent_image + noise
latent_image = self.model_sampling.calculate_input(torch.tensor([sigma_noise_augmentation], device=latent_image.device, dtype=latent_image.dtype), latent_image)
# print(latent_image.shape, latent_image)
# return latent_image
sigma = (sigma / (sigma + 1))
return latent_image / (1.0 - sigma)