Update utils.py

utils.py

@@ -212,7 +212,13 @@ def tokenize(
 
 
 def collate_fn(examples, tokenizer=None, padding=None, device=None):
-    batch = {k: [] for k in examples[0].keys()}
+    try:
+        keys = examples[0].keys()
+    except KeyError:
+        print(examples)
+    else:
+        batch = {k: [] for k in examples[0].keys()}
+        
 
     for example in examples:
         for k, v in example.items():