Fix Docs CI When Compile Error (#2323)
This commit is contained in:
@@ -220,19 +220,19 @@
|
||||
"metadata": {},
|
||||
"outputs": [],
|
||||
"source": [
|
||||
"# failed update with different parameter size\n",
|
||||
"# failed update with different parameter size or wrong name\n",
|
||||
"\n",
|
||||
"url = \"http://localhost:30010/update_weights_from_disk\"\n",
|
||||
"data = {\"model_path\": \"meta-llama/Llama-3.2-3B\"}\n",
|
||||
"data = {\"model_path\": \"meta-llama/Llama-3.2-1B-wrong\"}\n",
|
||||
"\n",
|
||||
"response = requests.post(url, json=data)\n",
|
||||
"response_json = response.json()\n",
|
||||
"print_highlight(response_json)\n",
|
||||
"assert response_json[\"success\"] is False\n",
|
||||
"assert response_json[\"message\"] == (\n",
|
||||
" \"Failed to update weights: The size of tensor a (2048) must match \"\n",
|
||||
" \"the size of tensor b (3072) at non-singleton dimension 1.\\n\"\n",
|
||||
" \"Rolling back to original weights.\"\n",
|
||||
" \"Failed to get weights iterator: \"\n",
|
||||
" \"meta-llama/Llama-3.2-1B-wrong\"\n",
|
||||
" \" (repository not found).\"\n",
|
||||
")"
|
||||
]
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user