minor changes

alibaba · Apr 12, 2024 · b285977 · b285977
1 parent dc30fdd
commit b285977
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git 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
@@ -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"]