vae must first be cuda

app.py

@@ -59,6 +59,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
     gc.collect()
     torch.cuda.empty_cache()
 
+    comparing_vae.to('cuda')
     comparing_unet.to('cuda')
     comparing_controlnet.to('cuda')
 
@@ -84,8 +85,6 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
     gc.collect()
     torch.cuda.empty_cache()
 
-    comparing_vae.to('cuda')
-
     out = comparing_vae.output_tensor_to_pil(comparing_vae.decode(out))
 
     comparing_vae.to('cpu')