diff --git a/rfdiffusion/inference/model_runners.py b/rfdiffusion/inference/model_runners.py index 9a33d98d..224d4c9c 100644 --- a/rfdiffusion/inference/model_runners.py +++ b/rfdiffusion/inference/model_runners.py @@ -237,7 +237,9 @@ def construct_contig(self, target_feats): Construct contig class describing the protein to be generated """ self._log.info(f'Using contig: {self.contig_conf.contigs}') - return ContigMap(target_feats, **self.contig_conf) + contig_map = ContigMap(target_feats, **self.contig_conf) + self._log.info(f'Sampled contig: {contig_map.sampled_mask}') + return contig_map def construct_denoiser(self, L, visible): """Make length-specific denoiser."""