zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-30 02:53:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix --encode-only

Browse Source
This commit is contained in:
Juarez Bochi 2023-12-18 11:19:44 -05:00
parent 83b68a5bdb
commit 0779417903
No known key found for this signature in database
GPG Key ID: 34CCBB77DC8BEBB6
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
t5/t5.py
View File

@ -394,8 +394,7 @@ if __name__ == "__main__":
if args.encode_only: if args.encode_only:
print("[INFO] Encoding with T5...", flush=True) print("[INFO] Encoding with T5...", flush=True)
print(args.prompt, flush=True) print(args.prompt, flush=True)
embeddings = model.wte(prompt) encoder_output = model.encode(prompt)
encoder_output = model.encoder(embeddings, mask=None)
print(encoder_output, flush=True) print(encoder_output, flush=True)
exit(0) exit(0)