add llamafy_internlm2

Former-commit-id: f1d7ca77b1
This commit is contained in:
hiyouga
2024-01-18 00:17:41 +08:00
parent 6a954cc075
commit a95dcac6f8
2 changed files with 123 additions and 1 deletions

View File

@@ -64,7 +64,8 @@ class Evaluator:
name=subject,
cache_dir=self.model_args.cache_dir,
download_mode=self.eval_args.download_mode,
token=self.model_args.hf_hub_token
token=self.model_args.hf_hub_token,
trust_remote_code=True
)
pbar.set_postfix_str(categorys[subject]["name"])
inputs, outputs, labels = [], [], []