Flux lora (#205)

2026-03-23 17:38:10 +00:00 · 2024-09-12 16:49:30 +08:00
parent a8cb4a21d1
commit c21ed1e478
2 changed files with 3 additions and 3 deletions
--- a/diffsynth/prompters/flux_prompter.py
+++ b/diffsynth/prompters/flux_prompter.py
@@ -49,8 +49,6 @@ class FluxPrompter(BasePrompter):
            truncation=True,
        ).input_ids.to(device)
        prompt_emb = text_encoder(input_ids)
-        prompt_emb = prompt_emb.reshape((1, prompt_emb.shape[0]*prompt_emb.shape[1], -1))
-
        return prompt_emb