Unverified Commit 786be44d authored by Chayenne's avatar Chayenne Committed by GitHub
Browse files

Fix Docs CI When Compile Error (#2323)

parent 2db44698
......@@ -19,7 +19,7 @@ compile:
echo "Executing $$nb"; \
jupyter nbconvert --to notebook --execute --inplace "$$nb" \
--ExecutePreprocessor.timeout=600 \
--ExecutePreprocessor.kernel_name=python3; \
--ExecutePreprocessor.kernel_name=python3 || exit 1; \
fi; \
done
......
......@@ -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",
")"
]
},
......
......@@ -329,7 +329,7 @@ async def encode_request(obj: EmbeddingReqInput, request: Request):
)
@app.api_route("/encode", methods=["POST", "PUT"])
@app.api_route("/classify", methods=["POST", "PUT"])
@time_func_latency
async def classify_request(obj: EmbeddingReqInput, request: Request):
"""Handle a reward model request. Now the arguments and return values are the same as embedding models."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment