Upload transformers_example.ipynb

transformers_example.ipynb

@@ -26,7 +26,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "ba9ec6c7",
    "metadata": {},
    "outputs": [],
@@ -35,12 +35,12 @@
     "MODEL = \"mdbr-leaf-ir\"\n",
     "\n",
     "tokenizer = AutoTokenizer.from_pretrained(MODEL)\n",
-    "model = AutoModel.from_pretrained(MODEL)"
+    "model = AutoModel.from_pretrained(MODEL, add_pooling_layer=False)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
    "id": "ebaf1a76",
    "metadata": {},
    "outputs": [
@@ -114,6 +114,14 @@
     "#  tensor([[0.6857, 0.4598],\n",
     "#          [0.4238, 0.5723]])"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "458cec94",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {