Update app.py
Browse files
app.py
CHANGED
|
@@ -37,15 +37,15 @@ def encode_text(prompt: str):
|
|
| 37 |
|
| 38 |
# Sortie Qwen : hidden_states = [1, seq_len, 4096]
|
| 39 |
outputs = model.model(**inputs, output_hidden_states=True)
|
| 40 |
-
hidden = outputs.hidden_states[-1]
|
| 41 |
|
| 42 |
# Projection FLUX2 Klein
|
| 43 |
-
projected = project_out(hidden) # [1, seq_len, 7680]
|
| 44 |
|
| 45 |
# Sauvegarde
|
| 46 |
fd, path = tempfile.mkstemp(suffix=".pt")
|
| 47 |
os.close(fd)
|
| 48 |
-
torch.save(
|
| 49 |
|
| 50 |
return path
|
| 51 |
|
|
|
|
| 37 |
|
| 38 |
# Sortie Qwen : hidden_states = [1, seq_len, 4096]
|
| 39 |
outputs = model.model(**inputs, output_hidden_states=True)
|
| 40 |
+
#hidden = outputs.hidden_states[-1]
|
| 41 |
|
| 42 |
# Projection FLUX2 Klein
|
| 43 |
+
#projected = project_out(hidden) # [1, seq_len, 7680]
|
| 44 |
|
| 45 |
# Sauvegarde
|
| 46 |
fd, path = tempfile.mkstemp(suffix=".pt")
|
| 47 |
os.close(fd)
|
| 48 |
+
torch.save(outputs.cpu(), path)
|
| 49 |
|
| 50 |
return path
|
| 51 |
|