Ver Fonte

gguf-py : fix SafetensorRemote return on undefined size (< 0) (#13841)

Beinsezii há 8 meses atrás
pai
commit
53ae30640e
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      gguf-py/gguf/utility.py

+ 1 - 1
gguf-py/gguf/utility.py

@@ -231,7 +231,7 @@ class SafetensorRemote:
         response.raise_for_status()
 
         # Get raw byte data
-        return response.content[:size]
+        return response.content[slice(size if size > -1 else None)]
 
     @classmethod
     def check_file_exist(cls, url: str) -> bool: