format

aibrix · Nov 8, 2024 · c2375b1 · c2375b1
1 parent 8d2c856
commit c2375b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/model_executor/model_loader/stream/utils.py b/vllm/model_executor/model_loader/stream/utils.py
@@ -73,7 +73,7 @@ def need_to_download(
 
 
 def read_to_bytes_io(content, chunk_size=None):
-    chunk_size = int(os.getenv("STREAM_READ_CHUNK_SIZE", 8388608))  # 8MB    
+    chunk_size = int(os.getenv("STREAM_READ_CHUNK_SIZE", 8388608))  # 8MB
     buf = content.read(chunk_size)
     _all_bufs = [buf]
 
@@ -84,6 +84,7 @@ def read_to_bytes_io(content, chunk_size=None):
     bytes_io.seek(0)
     return bytes_io
 
+
 def filter_suffix_files(files: List[str], suffix: str) -> List[str]:
     return [file for file in files if file.endswith(suffix)]