TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: change model size type to float (#126)

Browse Source
This commit is contained in:
Hayden
2025-02-06 12:02:00 +08:00
committed by GitHub
parent f6368fe20b
commit 17ab373b9c
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
py/download.py
View File

@@ -61,7 +61,7 @@ class TaskContent:
description: str description: str
downloadPlatform: str downloadPlatform: str
downloadUrl: str downloadUrl: str
sizeBytes: int sizeBytes: float
hashes: Optional[dict[str, str]] = None hashes: Optional[dict[str, str]] = None
def __init__(self, **kwargs): def __init__(self, **kwargs):
@@ -71,7 +71,7 @@ class TaskContent:
self.description = kwargs.get("description", None) self.description = kwargs.get("description", None)
self.downloadPlatform = kwargs.get("downloadPlatform", None) self.downloadPlatform = kwargs.get("downloadPlatform", None)
self.downloadUrl = kwargs.get("downloadUrl", None) self.downloadUrl = kwargs.get("downloadUrl", None)
self.sizeBytes = int(kwargs.get("sizeBytes", 0)) self.sizeBytes = float(kwargs.get("sizeBytes", 0))
self.hashes = kwargs.get("hashes", None) self.hashes = kwargs.get("hashes", None)
def to_dict(self): def to_dict(self):
@@ -382,7 +382,7 @@ async def download_model_file(
# When parsing model information from HuggingFace API, # When parsing model information from HuggingFace API,
# the file size was not found and needs to be obtained from the response header. # the file size was not found and needs to be obtained from the response header.
if total_size == 0: if total_size == 0:
total_size = int(response.headers.get("content-length", 0)) total_size = float(response.headers.get("content-length", 0))
task_content.sizeBytes = total_size task_content.sizeBytes = total_size
task_status.totalSize = total_size task_status.totalSize = total_size
set_task_content(task_id, task_content) set_task_content(task_id, task_content)