modified: app.py

app.py

@@ -38,8 +38,7 @@ def encode(init_image, torch_device, ae):
     init_image = init_image.unsqueeze(0) 
     init_image = init_image.to(torch_device)
     with torch.no_grad():
-        init_image = ae.encode(init_image.to()).to(torch.bfloat16)
-    init_image = init_image.to(torch_device)
+        init_image = ae.encode(init_image).to(torch.bfloat16)
     return init_image