(image, model, res=512)
| 358 | |
| 359 | |
| 360 | def encoder(image, model, res=512): |
| 361 | generator = torch.Generator().manual_seed(8888) |
| 362 | image = preprocess(image, res) |
| 363 | gpu_generator = torch.Generator(device=image.device) |
| 364 | gpu_generator.manual_seed(generator.initial_seed()) |
| 365 | return 0.18215 * model.vae.encode(image).latent_dist.sample(generator=gpu_generator) |
| 366 | |
| 367 | |
| 368 | @torch.no_grad() |
no test coverage detected