Update app.py
Browse files
app.py
CHANGED
|
@@ -14,10 +14,9 @@ text_encoder = Qwen2ForCausalLM.from_pretrained(
|
|
| 14 |
)
|
| 15 |
|
| 16 |
# Projection 896 -> 2048 pour FLUX.1-Schnell
|
| 17 |
-
proj_tokens = nn.Linear(896,
|
|
|
|
| 18 |
|
| 19 |
-
# Projection pooled 2048 -> 768 (obligatoire pour Schnell)
|
| 20 |
-
proj_pooled = nn.Linear(3072, 768)
|
| 21 |
|
| 22 |
def encode(prompt: str):
|
| 23 |
if not prompt or prompt.strip() == "":
|
|
|
|
| 14 |
)
|
| 15 |
|
| 16 |
# Projection 896 -> 2048 pour FLUX.1-Schnell
|
| 17 |
+
proj_tokens = nn.Linear(896, 2048)
|
| 18 |
+
proj_pooled = nn.Linear(2048, 768)
|
| 19 |
|
|
|
|
|
|
|
| 20 |
|
| 21 |
def encode(prompt: str):
|
| 22 |
if not prompt or prompt.strip() == "":
|