lea97338 commited on
Commit
1e689ad
·
verified ·
1 Parent(s): fe4b89c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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(projected.cpu(), path)
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