refactor tests

atoma-network · Sep 25, 2024 · a8de053 · a8de053
1 parent 17df8e0
commit a8de053
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backends/vllm/src/tests/test_config.toml b/backends/vllm/src/tests/test_config.toml
@@ -1,6 +1,6 @@
 [inference]
 api_key = ""
-cache_dir = "./cache/"
+cache_dir = "./"
 flush_storage = true
 model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 device_ids = [0]

diff --git a/backends/vllm/src/tests/test_config_enable_chunked_prefill.toml b/backends/vllm/src/tests/test_config_enable_chunked_prefill.toml
@@ -1,6 +1,6 @@
 [inference]
 api_key = ""
-cache_dir = "./cache/"
+cache_dir = "./"
 flush_storage = true
 model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
 device_ids = [0]