SIGN IN SIGN UP

Use checkpoint dtype to export llama (#1870)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/1870

Inspect the state_dict to get the dtype of the checkpoint.

Assumes that checkpoint only has one dtype, otherwise prints the mismatches.

add `--dtype-override` flag to explicitly specificy fp16, fp32. Remove -H flag.

Reviewed By: larryliu0820

Differential Revision: D53505038

fbshipit-source-id: 1073a155dcdc9c390981284094e3ab9be2c3b12d
L
Lucy Qiu committed
63489e88763a8f322a9486cd21db2f13a9e71305
Parent: 1363004
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 2/9/2024, 3:50:00 AM