make it work with transformers 4

Author: datanizing

transformers_multi_label_classification.ipynb

@@ -912,7 +912,7 @@
     "        self.l3 = torch.nn.Linear(768, 6)\n",
     "    \n",
     "    def forward(self, ids, mask, token_type_ids):\n",
-    "        _, output_1= self.l1(ids, attention_mask = mask, token_type_ids = token_type_ids)\n",
+    "        _, output_1= self.l1(ids, attention_mask = mask, token_type_ids = token_type_ids, return_dict=False)\n",
     "        output_2 = self.l2(output_1)\n",
     "        output = self.l3(output_2)\n",
     "        return output\n",