Clean up some debug lines.

This commit is contained in:
comfyanonymous 2025-01-16 04:24:39 -05:00
parent 9d8b6c1f46
commit 23289a6a5c

View File

@ -89,8 +89,8 @@ class CausalContinuousVideoTokenizer(nn.Module):
self.distribution = IdentityDistribution() # ContinuousFormulation[formulation_name].value()
num_parameters = sum(param.numel() for param in self.parameters())
logging.info(f"model={self.name}, num_parameters={num_parameters:,}")
logging.info(
logging.debug(f"model={self.name}, num_parameters={num_parameters:,}")
logging.debug(
f"z_channels={z_channels}, latent_channels={self.latent_channels}."
)