bug fix

KoichiYasuoka · Jul 4, 2024 · d0a7308 · d0a7308
1 parent ccc14e6
commit d0a7308
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/demo/2024-07-05/lzh-qwen-upos.ipynb b/demo/2024-07-05/lzh-qwen-upos.ipynb
@@ -21,7 +21,6 @@
         "!pip install transformers accelerate\n",
         "!test -d Xunzi-Qwen1.5-4B || env GIT_LFS_SKIP_SMUDGE=1 git clone --depth=1 https://www.modelscope.cn/Xunzillm4cc/Xunzi-Qwen1.5-4B.git\n",
         "from transformers import AutoTokenizer,AutoModelForTokenClassification,TokenClassificationPipeline\n",
-        "from transformers.utils import cached_file\n",
         "tkz=AutoTokenizer.from_pretrained(\"Xunzi-Qwen1.5-4B\")\n",
         "mdl=AutoModelForTokenClassification.from_pretrained(\"KoichiYasuoka/Xunzi-Qwen1.5-4B-upos\",trust_remote_code=True,device_map=\"auto\")\n",
         "class TCP(TokenClassificationPipeline):\n",