From d31be12590ed4fe2e2449fbeba36da19b8e81c87 Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Wed, 28 Feb 2024 10:33:05 +0800 Subject: [PATCH 1/2] fix angle-trainer --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4434d66..10185c1 100644 --- a/setup.py +++ b/setup.py @@ -42,7 +42,7 @@ tests_require=test_requirements, entry_points={ 'console_scripts': [ - 'angle-trainer = angle_emb.train_cli:__main__', + 'angle-trainer = angle_emb.train_cli:main', ], }, ) From 477c7e0d894a75392643c714c26f09b5e218a3b0 Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Wed, 28 Feb 2024 10:34:51 +0800 Subject: [PATCH 2/2] fix None error --- angle_emb/train_cli.py | 1 + 1 file changed, 1 insertion(+) diff --git a/angle_emb/train_cli.py b/angle_emb/train_cli.py index 2edec00..d5dba8c 100644 --- a/angle_emb/train_cli.py +++ b/angle_emb/train_cli.py @@ -171,6 +171,7 @@ def main(): 'fixed_teacher_name_or_path': args.fixed_teacher_name_or_path } if args.apply_tdmse: + trainer_kwargs = trainer_kwargs or {} trainer_kwargs = dict(trainer_kwargs, **{ 'apply_tdmse_kl': args.apply_tdmse_kl, 'tdmse_kl_temperature': args.tdmse_kl_temperature,