diff --git a/.gitignore b/.gitignore index 6c7580be..0ab9b639 100644 --- a/.gitignore +++ b/.gitignore @@ -44,3 +44,4 @@ test-encode-task.json docs/model_results/* model_results/* docs/model_results.json +speed_test.py diff --git a/src/seb/full_benchmark.py b/src/seb/full_benchmark.py index 9c376884..cfc732b3 100644 --- a/src/seb/full_benchmark.py +++ b/src/seb/full_benchmark.py @@ -60,7 +60,7 @@ def run_speed_benchmark( Run the speed benchmark. """ models: list[EmbeddingModel] = get_all_models() - tasks = [CPUSpeedTask(), GPUSpeedTask()] # type: ignore + tasks = [CPUSpeedTask()] # type: ignore if use_cache: logger.warn(