fix: removed GPU test from speed test

at least for the moment
KennethEnevoldsen · Jan 31, 2024 · a7600bd · a7600bd
1 parent ec07680
commit a7600bd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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(