[🤗] Safely get model_name from opt

parlai/agents/hugging_face/dict.py

@@ -140,7 +140,7 @@ def get_tokenizer(self, opt):
         """
         Instantiate tokenizer.
         """
-        if opt["model_name"]:
+        if opt.get("model_name"):
             fle_key = opt["model_name"]
         else:
             model_sz = opt["gpt2_size"]

parlai/agents/hugging_face/gpt2.py

@@ -58,7 +58,7 @@ def __init__(self, opt, dict):
 
     def _init_from_pretrained(self, opt):
         # load model
-        if opt["model_name"]:
+        if opt.get("model_name"):
             fle_key = opt["model_name"]
         else:
             model_sz = opt["gpt2_size"]

parlai/tasks/lccc/build.py

@@ -9,7 +9,7 @@
         'https://cloud.tsinghua.edu.cn/f/f131a4d259184566a29c/?dl=1',
         'LCCC.zip',
         'f5203511cd8d6a608008af0aa290aa516d983abc16aa510471e3c4ee6bca7886',
-    ),
+    )
 ]
 
 

parlai/tasks/task_list.py

@@ -1554,9 +1554,7 @@
         "display_name": "LCCC",
         "task": "lccc",
         "tags": ["ChitChat"],
-        "description": (
-            "Large-scale cleaned Chinese conversation dataset."
-        ),
+        "description": ("Large-scale cleaned Chinese conversation dataset."),
         "links": {
             "arXiv": "https://arxiv.org/pdf/2008.03946",
             "website": "https://github.com/thu-coai/CDial-GPT",