wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-21 03:50:50 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

fixed hash calculation during model loading in ComfyUI

Browse Source
This commit is contained in:
bigcat88 2025-09-17 13:25:56 +03:00
parent 621faaa195
commit 5b6810a2c6
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
app/database/services/content.py
View File

@ -33,14 +33,18 @@ async def check_fs_asset_exists_quick(
size_bytes: Optional[int] = None, size_bytes: Optional[int] = None,
mtime_ns: Optional[int] = None, mtime_ns: Optional[int] = None,
) -> bool: ) -> bool:
"""Return True if a cache row exists for this absolute path and (optionally) mtime/size match.""" """Returns True if we already track this absolute path with a HASHED asset and the cached mtime/size match."""
locator = os.path.abspath(file_path) locator = os.path.abspath(file_path)
stmt = ( stmt = (
sa.select(sa.literal(True)) sa.select(sa.literal(True))
.select_from(AssetCacheState) .select_from(AssetCacheState)
.join(Asset, Asset.id == AssetCacheState.asset_id) .join(Asset, Asset.id == AssetCacheState.asset_id)
.where(AssetCacheState.file_path == locator) .where(
AssetCacheState.file_path == locator,
Asset.hash.isnot(None),
AssetCacheState.needs_verify.is_(False),
)
.limit(1) .limit(1)
) )
@ -49,12 +53,9 @@ async def check_fs_asset_exists_quick(
conds.append(AssetCacheState.mtime_ns == int(mtime_ns)) conds.append(AssetCacheState.mtime_ns == int(mtime_ns))
if size_bytes is not None: if size_bytes is not None:
conds.append(sa.or_(Asset.size_bytes == 0, Asset.size_bytes == int(size_bytes))) conds.append(sa.or_(Asset.size_bytes == 0, Asset.size_bytes == int(size_bytes)))
if conds: if conds:
stmt = stmt.where(*conds) stmt = stmt.where(*conds)
return (await session.execute(stmt)).first() is not None
row = (await session.execute(stmt)).first()
return row is not None
async def seed_from_path( async def seed_from_path(