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

add migration_11; fix typo

Browse Source
This commit is contained in:
Lincoln Stein 2024-05-27 22:40:01 -04:00
parent 34e1eb19f9
commit cd12ca6e85
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/services/model_install/model_install_default.py
View File

@ -750,8 +750,8 @@ class ModelInstallService(ModelInstallServiceBase):
self._download_cache[multifile_job.id] = install_job self._download_cache[multifile_job.id] = install_job
install_job._download_job = multifile_job install_job._download_job = multifile_job
files_string = "file" if len(remote_files) == 1 else "file" files_string = "file" if len(remote_files) == 1 else "files"
self._logger.info(f"Queuing model install: {source} ({len(remote_files)} {files_string})") self._logger.info(f"Queueing model install: {source} ({len(remote_files)} {files_string})")
self._logger.debug(f"remote_files={remote_files}") self._logger.debug(f"remote_files={remote_files}")
self._download_queue.submit_multifile_download(multifile_job) self._download_queue.submit_multifile_download(multifile_job)
return install_job return install_job
@ -828,6 +828,7 @@ class ModelInstallService(ModelInstallServiceBase):
else: else:
# update sizes # update sizes
install_job.bytes = sum(x.bytes for x in download_job.download_parts) install_job.bytes = sum(x.bytes for x in download_job.download_parts)
install_job.total_bytes = sum(x.total_bytes for x in download_job.download_parts)
install_job.download_parts = download_job.download_parts install_job.download_parts = download_job.download_parts
self._signal_job_downloading(install_job) self._signal_job_downloading(install_job)

2
invokeai/app/services/shared/sqlite/sqlite_util.py
View File

@ -13,6 +13,7 @@ from invokeai.app.services.shared.sqlite_migrator.migrations.migration_7 import
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_8 import build_migration_8 from invokeai.app.services.shared.sqlite_migrator.migrations.migration_8 import build_migration_8
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_9 import build_migration_9 from invokeai.app.services.shared.sqlite_migrator.migrations.migration_9 import build_migration_9
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_10 import build_migration_10 from invokeai.app.services.shared.sqlite_migrator.migrations.migration_10 import build_migration_10
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_11 import build_migration_11
from invokeai.app.services.shared.sqlite_migrator.sqlite_migrator_impl import SqliteMigrator from invokeai.app.services.shared.sqlite_migrator.sqlite_migrator_impl import SqliteMigrator
@ -43,6 +44,7 @@ def init_db(config: InvokeAIAppConfig, logger: Logger, image_files: ImageFileSto
migrator.register_migration(build_migration_8(app_config=config)) migrator.register_migration(build_migration_8(app_config=config))
migrator.register_migration(build_migration_9()) migrator.register_migration(build_migration_9())
migrator.register_migration(build_migration_10()) migrator.register_migration(build_migration_10())
migrator.register_migration(build_migration_11(app_config=config, logger=logger))
migrator.run_migrations() migrator.run_migrations()
return db return db