From 646add55f14ef287c39dcdced0719503437e2217 Mon Sep 17 00:00:00 2001 From: degenfabian Date: Tue, 19 Aug 2025 02:57:58 +0200 Subject: [PATCH 1/2] updating loading in t5 demo to use transformer bridge --- demos/T5.ipynb | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/demos/T5.ipynb b/demos/T5.ipynb index 1a2de7ba5..9188862e5 100644 --- a/demos/T5.ipynb +++ b/demos/T5.ipynb @@ -77,7 +77,7 @@ }, { "cell_type": "code", - "execution_count": 6, + "execution_count": null, "metadata": {}, "outputs": [ { @@ -110,10 +110,11 @@ "import torch\n", "\n", "from transformers import AutoTokenizer\n", - "from transformer_lens import HookedEncoderDecoder\n", + "from transformer_lens.model_bridge import TransformerBridge\n", "\n", - "model_name = \"t5-small\"\n", - "model = HookedEncoderDecoder.from_pretrained(model_name)\n", + "model_name = \"google-t5/t5-small\"\n", + "model = TransformerBridge.boot_transformers(model_name)\n", + "model.enable_compatibility_mode()\n", "tokenizer = AutoTokenizer.from_pretrained(model_name)" ] }, From 5c9490488af48c1c456b094a3e7fae7007dbb706 Mon Sep 17 00:00:00 2001 From: degenfabian Date: Tue, 19 Aug 2025 18:01:54 +0200 Subject: [PATCH 2/2] add T5 demo to CI --- .github/workflows/checks.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/checks.yml b/.github/workflows/checks.yml index 4de51026c..d519d30a4 100644 --- a/.github/workflows/checks.yml +++ b/.github/workflows/checks.yml @@ -159,7 +159,7 @@ jobs: # - "No_Position_Experiment" - "Othello_GPT" - "Patchscopes_Generation_Demo" - # - "T5" + - "T5" steps: - uses: actions/checkout@v3 - name: Install Poetry