(self, latent, tiled=False, tile_size=64, tile_stride=32)
| 54 | |
| 55 | |
| 56 | def decode_image(self, latent, tiled=False, tile_size=64, tile_stride=32): |
| 57 | image = self.vae_decoder(latent.to(self.device), tiled=tiled, tile_size=tile_size, tile_stride=tile_stride) |
| 58 | image = self.vae_output_to_image(image) |
| 59 | return image |
| 60 | |
| 61 | |
| 62 | def encode_prompt(self, prompt, positive=True, t5_sequence_length=77): |
no test coverage detected