Fix loading Phi-3 (#48)
This commit is contained in:
committed by
GitHub
parent
e55a0738aa
commit
aabef6b341
@@ -64,7 +64,7 @@
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"execution_count": null,
|
||||
"metadata": {
|
||||
"colab": {
|
||||
"base_uri": "https://localhost:8080/",
|
||||
@@ -278,7 +278,7 @@
|
||||
" \"microsoft/Phi-3-mini-4k-instruct\",\n",
|
||||
" device_map=\"cuda\",\n",
|
||||
" torch_dtype=\"auto\",\n",
|
||||
" trust_remote_code=True,\n",
|
||||
" trust_remote_code=False,\n",
|
||||
")\n",
|
||||
"tokenizer = AutoTokenizer.from_pretrained(\"microsoft/Phi-3-mini-4k-instruct\")"
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user