Fix: unsqueeze prompt embeddings

app.py

@@ -66,8 +66,8 @@ def _load_all():
 
     print("[init] Loading empty prompt cache...")
     data = np.load(EMPTY_PROMPT_CACHE, allow_pickle=True)
-    embeds = torch.from_numpy(data["embeds"])
-    masks = torch.from_numpy(data["masks"])
+    embeds = torch.from_numpy(data["embeds"]).unsqueeze(0)
+    masks = torch.from_numpy(data["masks"]).unsqueeze(0)
 
     print("[init] Ready.")
     return dit_session, vae, embeds, masks