guess latent channel count based on encoder filename
(self, encoder_path)
| 82 | return self.decoder(z).mul(2).sub(1) |
| 83 | |
| 84 | def guess_latent_channels(self, encoder_path): |
| 85 | """guess latent channel count based on encoder filename""" |
| 86 | if "taef1" in encoder_path: |
| 87 | return 16 |
| 88 | if "taesd3" in encoder_path: |
| 89 | return 16 |
| 90 | return 4 |
| 91 | |
| 92 | @staticmethod |
| 93 | def scale_latents(x): |