Update app.py

app.py

@@ -780,7 +780,7 @@ def generate_image(
             init_image = init_image[..., : 16 * (h // 16), : 16 * (w // 16)]
             height = init_image.shape[-2]
             width = init_image.shape[-1]
-        init_image = ae.encode(init_image.to(torch_device)).latent_dist.sample()
+        init_image = ae.encode(init_image.to(torch_device).to(torch.bfloat16)).latent_dist.sample()
         init_image =  (init_image - ae.config.shift_factor) * ae.config.scaling_factor
 
     generator = torch.Generator(device=device).manual_seed(seed)