Skip to content

Commit

Permalink
merging main
Browse files Browse the repository at this point in the history
  • Loading branch information
rlundeen2 committed Nov 18, 2024
2 parents cd954ca + e4cb06f commit 85a707d
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 61 deletions.
2 changes: 1 addition & 1 deletion doc/_toc.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,6 @@ chapters:
- file: code/orchestrators/pku_safe_rlhf_testing
- file: code/orchestrators/seclists_bias_testing
- file: code/orchestrators/tree_of_attacks_with_pruning
- file: code/orchestrators/use_huggingface_chat_target
- file: code/orchestrators/violent_durian
- file: code/orchestrators/xstest_bias_testing
- file: code/targets/0_prompt_targets
Expand All @@ -56,6 +55,7 @@ chapters:
- file: code/targets/7_http_target
- file: code/targets/open_ai_completions
- file: code/targets/prompt_shield_target
- file: code/targets/use_huggingface_chat_target
- file: code/converters/0_converters
sections:
- file: code/converters/1_llm_converters
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"id": "940f8d8a",
"metadata": {
"execution": {
Expand All @@ -47,64 +47,7 @@
"shell.execute_reply": "2024-11-11T22:43:23.862727Z"
}
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Running model: HuggingFaceTB/SmolLM-135M-Instruct\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"Average response time for HuggingFaceTB/SmolLM-135M-Instruct: 37.12 seconds\n",
"\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"\u001b[22m\u001b[39mConversation ID: 5223e15e-f21c-4d15-88af-8c02d6558182\n",
"\u001b[1m\u001b[34muser: What is 4*4? Give me the solution.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"\u001b[22m\u001b[33massistant: What a great question!\n",
"\n",
"The number 4*4 is a special number because it can be expressed as a product of two numbers,\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"\u001b[22m\u001b[39mConversation ID: b0238d3e-ce2e-48c3-a5e1-eaebf2c58e6f\n",
"\u001b[1m\u001b[34muser: What is 3*3? Give me the solution.\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"\u001b[22m\u001b[33massistant: What a great question!\n",
"\n",
"The number 3*3 is a fascinating number that has been a subject of fascination for mathematicians and computer scientists for\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"HuggingFaceTB/SmolLM-135M-Instruct: 37.12 seconds\n"
]
}
],
"outputs": [],
"source": [
"import time\n",
"from pyrit.prompt_target import HuggingFaceChatTarget\n",
Expand Down Expand Up @@ -166,7 +109,7 @@
"kernelspec": {
"display_name": "pyrit-dev",
"language": "python",
"name": "pyrit-dev"
"name": "python3"
},
"language_info": {
"codemirror_mode": {
Expand Down

0 comments on commit 85a707d

Please sign in to comment.