From b28597795e9e2a9d895cf63b80529de5293c613c Mon Sep 17 00:00:00 2001 From: qbc Date: Fri, 12 Apr 2024 15:29:16 +0800 Subject: [PATCH] minor changes --- federatedscope/llm/eval/eval_for_finance/eval.py | 3 +++ federatedscope/llm/eval/eval_for_mmlu/eval.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/federatedscope/llm/eval/eval_for_finance/eval.py b/federatedscope/llm/eval/eval_for_finance/eval.py index 7d0ebe39f..92711697f 100644 --- a/federatedscope/llm/eval/eval_for_finance/eval.py +++ b/federatedscope/llm/eval/eval_for_finance/eval.py @@ -12,6 +12,9 @@ from federatedscope.llm.misc.fschat import FSChatBot from federatedscope.core.data.utils import download_url +# import torch._dynamo +# torch._dynamo.config.suppress_errors = True + transformers.logging.set_verbosity(40) choices = ["A", "B", "C", "D"] diff --git a/federatedscope/llm/eval/eval_for_mmlu/eval.py b/federatedscope/llm/eval/eval_for_mmlu/eval.py index 68a59be87..97e897e68 100644 --- a/federatedscope/llm/eval/eval_for_mmlu/eval.py +++ b/federatedscope/llm/eval/eval_for_mmlu/eval.py @@ -16,6 +16,9 @@ from federatedscope.core.data.utils import download_url import tarfile +# import torch._dynamo +# torch._dynamo.config.suppress_errors = True + transformers.logging.set_verbosity(40) choices = ["A", "B", "C", "D"]