Xuechen Li 7fcd3e6a04 map custom model state_dict back to huggingface format (#465) ... * fix name. * set inv function. * add map back function. * handle gqa. * add type annotation to avoid confusion. * fix docstr. * test inverse remap logic.		2023-08-18 20:51:39 -07:00
..
__init__.py	Add __init__.py files to subdirectories for installation	2022-11-17 16:55:44 -08:00
bert.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
falcon.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
gpt_neox.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
gpt.py	map custom model state_dict back to huggingface format (#465)	2023-08-18 20:51:39 -07:00
gptj.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
llama.py	map custom model state_dict back to huggingface format (#465)	2023-08-18 20:51:39 -07:00
opt.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00
vit.py	Run isort and black on python files	2023-08-18 14:22:11 -07:00