derek-thomas
/

tgi-notebooks-optimization

Model card Files Files and versions

derek-thomas commited on Jun 2, 2024

Commit

262d1ee

·

verified ·

1 Parent(s): 0e1f8ee

Adding 8 to the batch sizes

Files changed (1) hide show

01-tgi-ie-benchmark.ipynb +1 -1

01-tgi-ie-benchmark.ipynb CHANGED Viewed

@@ -75,7 +75,7 @@
     "\n",
     "# Simulation\n",
     "RESULTS_DIR = proj_dir/'tgi_bench_results'/INSTANCE_TYPE\n",
-    "tgi_bss = [16, 24, 32, 40, 48, 56, 64]"
    ]
   },
   {

     "\n",
     "# Simulation\n",
     "RESULTS_DIR = proj_dir/'tgi_bench_results'/INSTANCE_TYPE\n",
+    "tgi_bss = [8, 16, 24, 32, 40, 48, 56, 64]"
    ]
   },
   {