update

app.py

@@ -72,11 +72,12 @@ def init():
         tokenizer_2=tokenizer_2,
         vae=vae,
         transformer=None,
-    ).to(DEVICE)
+    )
     pipe.text_encoder_2 = text_encoder_2
     pipe.transformer = transformer
     if DEVICE == "cuda":
         pipe.enable_model_cpu_offload()
+    pipe.to(DEVICE)
 
 #    generator = torch.Generator().manual_seed(12345)
 #    image = pipe(