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

fixing issue where default board did not return images

Browse Source
This commit is contained in:
Stefan Tobler 2024-01-14 01:33:43 -05:00 committed by Brandon Rising
parent c4cdaaf4dd
commit 1e00b9760a
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
invokeai/app/services/bulk_download/bulk_download_default.py
View File

@ -10,7 +10,7 @@ from invokeai.app.services.bulk_download.bulk_download_common import (
BulkDownloadTargetException, BulkDownloadTargetException,
) )
from invokeai.app.services.events.events_base import EventServiceBase from invokeai.app.services.events.events_base import EventServiceBase
from invokeai.app.services.image_records.image_records_common import ImageRecordNotFoundException from invokeai.app.services.image_records.image_records_common import ImageCategory, ImageRecordNotFoundException
from invokeai.app.services.invoker import Invoker from invokeai.app.services.invoker import Invoker
from .bulk_download_base import BulkDownloadBase from .bulk_download_base import BulkDownloadBase
@ -67,7 +67,17 @@ class BulkDownloadService(BulkDownloadBase):
try: try:
board_name: str = "" board_name: str = ""
if board_id: if board_id:
image_names = invoker.services.board_image_records.get_all_board_image_names_for_board(board_id) # -1 is the default value for limit, which means no limit, is_intermediate only gives us completed images
image_names = [
img.image_name
for img in invoker.services.images.get_many(
offset=0,
limit=-1,
board_id=board_id,
is_intermediate=False,
categories=[ImageCategory.GENERAL],
).items
]
if board_id == "none": if board_id == "none":
board_id = "Uncategorized" board_id = "Uncategorized"
board_name = "Uncategorized" board_name = "Uncategorized"