Skip to content

Commit 568a484

Browse files
blessedcoolanthipsterusername
authored andcommitted
fix: other recursive imports
1 parent b1e56e2 commit 568a484

File tree

3 files changed

+10
-14
lines changed

3 files changed

+10
-14
lines changed

invokeai/app/services/download/download_default.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import traceback
99
from pathlib import Path
1010
from queue import Empty, PriorityQueue
11-
from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional, Set
11+
from typing import Any, Dict, List, Literal, Optional, Set
1212

1313
import requests
1414
from pydantic.networks import AnyHttpUrl
@@ -34,9 +34,6 @@
3434
UnknownJobIDException,
3535
)
3636

37-
if TYPE_CHECKING:
38-
from invokeai.app.services.events.events_base import EventServiceBase
39-
4037
# Maximum number of bytes to download during each call to requests.iter_content()
4138
DOWNLOAD_CHUNK_SIZE = 100000
4239

invokeai/app/services/events/events_base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434
if TYPE_CHECKING:
3535
from invokeai.app.invocations.baseinvocation import BaseInvocation, BaseInvocationOutput
3636
from invokeai.app.services.download.download_base import DownloadJob
37-
from invokeai.app.services.events.events_common import EventBase
3837
from invokeai.app.services.model_install.model_install_common import ModelInstallJob
3938
from invokeai.app.services.session_processor.session_processor_common import ProgressImage
4039
from invokeai.app.services.session_queue.session_queue_common import (

invokeai/app/services/model_install/model_install_default.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from queue import Empty, Queue
1010
from shutil import copyfile, copytree, move, rmtree
1111
from tempfile import mkdtemp
12-
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union
12+
from typing import Any, Dict, List, Optional, Tuple, Type, Union
1313

1414
import torch
1515
import yaml
@@ -60,9 +60,6 @@
6060

6161
TMPDIR_PREFIX = "tmpinstall_"
6262

63-
if TYPE_CHECKING:
64-
from invokeai.app.services.events.events_base import EventServiceBase
65-
6663

6764
class ModelInstallService(ModelInstallServiceBase):
6865
"""class for InvokeAI model installation."""
@@ -412,11 +409,14 @@ def _remote_files_from_source(
412409
if isinstance(source, HFModelSource):
413410
metadata = HuggingFaceMetadataFetch(self._session).from_id(source.repo_id, source.variant)
414411
assert isinstance(metadata, ModelMetadataWithFiles)
415-
return metadata.download_urls(
416-
variant=source.variant or self._guess_variant(),
417-
subfolder=source.subfolder,
418-
session=self._session,
419-
), metadata
412+
return (
413+
metadata.download_urls(
414+
variant=source.variant or self._guess_variant(),
415+
subfolder=source.subfolder,
416+
session=self._session,
417+
),
418+
metadata,
419+
)
420420

421421
if isinstance(source, URLModelSource):
422422
try:

0 commit comments

Comments
 (0)