InvokeAI 6.14.2__tar.gz → 6.14.4__tar.gz

{invokeai-6.14.2 → invokeai-6.14.4}/InvokeAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: InvokeAI
-Version: 6.14.2
+Version: 6.14.4
 Summary: A full-featured AI-assisted image generation environment designed for creatives and enthusiasts.
 Author-email: Invoke <support@invoke.ai>
 License:                                  Apache License
@@ -234,6 +234,7 @@ Requires-Dist: torchvision
 Requires-Dist: transformers<5.6,>=5.5
 Requires-Dist: fastapi-events
 Requires-Dist: fastapi<0.142,>=0.141.1
+Requires-Dist: starlette<1.8
 Requires-Dist: huggingface-hub
 Requires-Dist: networkx
 Requires-Dist: pydantic-settings

{invokeai-6.14.2 → invokeai-6.14.4}/InvokeAI.egg-info/SOURCES.txt

@@ -982,12 +982,12 @@ invokeai/frontend/install/__init__.py
 invokeai/frontend/install/import_images.py
 invokeai/frontend/web/__init__.py
 invokeai/frontend/web/dist/index.html
-invokeai/frontend/web/dist/assets/App-DvlZKGz7.js
+invokeai/frontend/web/dist/assets/App-BDKBD7rc.js
 invokeai/frontend/web/dist/assets/App-tn1JZtBo.css
 invokeai/frontend/web/dist/assets/browser-ponyfill-BpMPqZAt.js
 invokeai/frontend/web/dist/assets/chunk-jRWAZmH_.js
 invokeai/frontend/web/dist/assets/denoising-strength-4mznP2tj.png
-invokeai/frontend/web/dist/assets/index-BuK6cFYB.js
+invokeai/frontend/web/dist/assets/index-B-mZDT6A.js
 invokeai/frontend/web/dist/assets/inter-cyrillic-ext-wght-normal-BOeWTOD4.woff2
 invokeai/frontend/web/dist/assets/inter-cyrillic-wght-normal-DqGufNeO.woff2
 invokeai/frontend/web/dist/assets/inter-greek-ext-wght-normal-DlzME5K_.woff2

{invokeai-6.14.2 → invokeai-6.14.4}/InvokeAI.egg-info/requires.txt

@@ -18,6 +18,7 @@ torchvision
 transformers<5.6,>=5.5
 fastapi-events
 fastapi<0.142,>=0.141.1
+starlette<1.8
 huggingface-hub
 networkx
 pydantic-settings

{invokeai-6.14.2 → invokeai-6.14.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: InvokeAI
-Version: 6.14.2
+Version: 6.14.4
 Summary: A full-featured AI-assisted image generation environment designed for creatives and enthusiasts.
 Author-email: Invoke <support@invoke.ai>
 License:                                  Apache License
@@ -234,6 +234,7 @@ Requires-Dist: torchvision
 Requires-Dist: transformers<5.6,>=5.5
 Requires-Dist: fastapi-events
 Requires-Dist: fastapi<0.142,>=0.141.1
+Requires-Dist: starlette<1.8
 Requires-Dist: huggingface-hub
 Requires-Dist: networkx
 Requires-Dist: pydantic-settings

{invokeai-6.14.2 → invokeai-6.14.4}/invokeai/app/services/model_install/model_install_default.py

@@ -317,6 +317,9 @@ class ModelInstallService(ModelInstallServiceBase):
 
     def _resume_remote_download(self, job: ModelInstallJob) -> None:
         job.status = InstallStatus.WAITING
+        # Sources whose partial file has vanished. _enqueue_remote_download replaces job.download_parts
+        # with fresh parts, so the flag must be carried onto them or the resume response loses it.
+        restarted_from_scratch: set[str] = set()
         if job.download_parts:
             for part in job.download_parts:
                 if part.complete or part.bytes <= 0:
@@ -328,6 +331,7 @@ class ModelInstallService(ModelInstallServiceBase):
                     part.bytes = 0
                     part.resume_from_scratch = True
                     part.resume_message = "Partial file missing. Restarted download from the beginning."
+                    restarted_from_scratch.add(str(part.source))
             job.bytes = sum(p.bytes for p in job.download_parts)
         remote_files, metadata = self._remote_files_from_source(job.source)
         subfolders = job.source.subfolders if isinstance(job.source, HFModelSource) else []
@@ -340,6 +344,7 @@ class ModelInstallService(ModelInstallServiceBase):
             subfolder=job.source.subfolder if isinstance(job.source, HFModelSource) and len(subfolders) <= 1 else None,
             subfolders=subfolders if len(subfolders) > 1 else None,
             resume_metadata=job._resume_metadata,
+            restarted_from_scratch=restarted_from_scratch,
         )
 
     @property
@@ -1340,6 +1345,7 @@ class ModelInstallService(ModelInstallServiceBase):
         subfolders: Optional[List[Path]] = None,
         resume_metadata: Optional[dict] = None,
         clear_partials: bool = False,
+        restarted_from_scratch: Optional[set[str]] = None,
     ) -> ModelInstallJob:
         job.source_metadata = metadata
         job.local_path = destdir
@@ -1382,9 +1388,15 @@ class ModelInstallService(ModelInstallServiceBase):
                 part.final_url = meta.get("final_url") or part.final_url
                 if meta.get("download_path"):
                     part.download_path = Path(meta.get("download_path"))
+        if restarted_from_scratch:
+            for part in multifile_job.download_parts:
+                if str(part.source) in restarted_from_scratch:
+                    part.resume_from_scratch = True
+                    part.resume_message = "Partial file missing. Restarted download from the beginning."
         with self._lock:
             self._download_cache[multifile_job.id] = job
         job._multifile_job = multifile_job
+        job.download_parts = multifile_job.download_parts
 
         self._write_install_marker(job, status=InstallStatus.WAITING)
         files_string = "file" if len(remote_files) == 1 else "files"