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