Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(nodes): fix off-by-one page count error

Browse Source
This commit is contained in:
psychedelicious 2023-04-04 16:38:37 +10:00
parent 406039426a
commit de0df4945d
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/image_storage.py
View File

@ -95,7 +95,10 @@ class DiskImageStorage(ImageStorageBase):
count = len(all_images) count = len(all_images)
page_of_images = all_images[page * per_page : (page + 1) * per_page] page_of_images = all_images[page * per_page : (page + 1) * per_page]
page_count = int(count / per_page) + 1
page_count_trunc = int(count / per_page)
page_count_mod = count % per_page
page_count = page_count_trunc if page_count_mod == 0 else page_count_trunc + 1
return PaginatedResults[ImageField]( return PaginatedResults[ImageField](
items=page_of_images, items=page_of_images,

6
invokeai/app/services/sqlite.py
View File

@ -106,10 +106,12 @@ class SqliteItemStorage(ItemStorageABC, Generic[T]):
finally: finally:
self._lock.release() self._lock.release()
pageCount = int(count / per_page) + 1 page_count_trunc = int(count / per_page)
page_count_mod = count % per_page
page_count = page_count_trunc if page_count_mod == 0 else page_count_trunc + 1
return PaginatedResults[T]( return PaginatedResults[T](
items=items, page=page, pages=pageCount, per_page=per_page, total=count items=items, page=page, pages=page_count, per_page=per_page, total=count
) )
def search( def search(