diff --git a/python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py b/python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py index 2a159e493..70b2203e0 100644 --- a/python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py +++ b/python/sglang/srt/mem_cache/storage/hf3fs/storage_hf3fs.py @@ -524,7 +524,9 @@ class HiCacheHF3FS(HiCacheStorage): flat = not self.is_zero_copy values = ( [ - self.mem_pool_host.get_data_page(host_indices[i * page_num], flat=flat) + self.mem_pool_host.get_data_page( + host_indices[i * self.mem_pool_host.page_size], flat=flat + ) for i in range(page_num) ] if self.is_zero_copy @@ -574,7 +576,9 @@ class HiCacheHF3FS(HiCacheStorage): # host_indices to kv_buffer flat = not self.is_zero_copy values = [ - self.mem_pool_host.get_data_page(host_indices[i * page_num], flat=flat) + self.mem_pool_host.get_data_page( + host_indices[i * self.mem_pool_host.page_size], flat=flat + ) for i in range(page_num) ]