Fixing naming
Browse files
01-tgi-ie-benchmark.ipynb
CHANGED
@@ -68,13 +68,13 @@
|
|
68 |
"outputs": [],
|
69 |
"source": [
|
70 |
"# Endpoint\n",
|
71 |
-
"ENDPOINT_NAME=\"
|
72 |
"NAMESPACE = 'hf-test-lab'\n",
|
73 |
"MODEL = 'meta-llama/Meta-Llama-3-8B-Instruct'\n",
|
74 |
"INSTANCE_TYPE = 'nvidia-a100_2'\n",
|
75 |
"\n",
|
76 |
"# Simulation\n",
|
77 |
-
"RESULTS_DIR = proj_dir/'
|
78 |
"tgi_bss = [8, 16, 24, 32, 40, 48, 56, 64]"
|
79 |
]
|
80 |
},
|
|
|
68 |
"outputs": [],
|
69 |
"source": [
|
70 |
"# Endpoint\n",
|
71 |
+
"ENDPOINT_NAME=\"tgi-benchmark-sp\"\n",
|
72 |
"NAMESPACE = 'hf-test-lab'\n",
|
73 |
"MODEL = 'meta-llama/Meta-Llama-3-8B-Instruct'\n",
|
74 |
"INSTANCE_TYPE = 'nvidia-a100_2'\n",
|
75 |
"\n",
|
76 |
"# Simulation\n",
|
77 |
+
"RESULTS_DIR = proj_dir/'tgi_benchmark_results'/INSTANCE_TYPE\n",
|
78 |
"tgi_bss = [8, 16, 24, 32, 40, 48, 56, 64]"
|
79 |
]
|
80 |
},
|