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 python static checks.

Browse Source
This commit is contained in:
Ryan Dick 2023-09-14 16:48:47 -04:00
parent e02a557454
commit c2f074dc2f
5 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/app/invocations/ip_adapter.py
View File

@ -1,5 +1,3 @@
from typing import Literal
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from invokeai.app.invocations.baseinvocation import ( from invokeai.app.invocations.baseinvocation import (

3
invokeai/backend/ip_adapter/ip_adapter.py
View File

@ -1,7 +1,6 @@
# copied from https://github.com/tencent-ailab/IP-Adapter (Apache License 2.0) # copied from https://github.com/tencent-ailab/IP-Adapter (Apache License 2.0)
# and modified as needed # and modified as needed
import os
from contextlib import contextmanager from contextlib import contextmanager
from typing import Optional, Union from typing import Optional, Union
@ -18,8 +17,6 @@ from diffusers.models import UNet2DConditionModel
from PIL import Image from PIL import Image
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
from invokeai.backend.model_management.models.base import calc_model_size_by_data
from .attention_processor import AttnProcessor, IPAttnProcessor from .attention_processor import AttnProcessor, IPAttnProcessor
from .resampler import Resampler from .resampler import Resampler

5
invokeai/backend/model_management/model_probe.py
View File

@ -8,7 +8,6 @@ import torch
from diffusers import ConfigMixin, ModelMixin from diffusers import ConfigMixin, ModelMixin
from picklescan.scanner import scan_file_path from picklescan.scanner import scan_file_path
from invokeai.backend.model_management.models import BaseModelType
from invokeai.backend.model_management.models.ip_adapter import IPAdapterModelFormat from invokeai.backend.model_management.models.ip_adapter import IPAdapterModelFormat
from .models import ( from .models import (
@ -512,7 +511,9 @@ class ControlNetFolderProbe(FolderProbeBase):
else ( else (
BaseModelType.StableDiffusion2 BaseModelType.StableDiffusion2
if dimension == 1024 if dimension == 1024
else BaseModelType.StableDiffusionXL if dimension == 2048 else None else BaseModelType.StableDiffusionXL
if dimension == 2048
else None
) )
) )
if not base_model: if not base_model:

3
invokeai/backend/model_management/models/ip_adapter.py
View File

@ -1,7 +1,7 @@
import os import os
import typing import typing
from enum import Enum from enum import Enum
from typing import Any, Literal, Optional from typing import Literal, Optional
import torch import torch
@ -17,7 +17,6 @@ from invokeai.backend.model_management.models.base import (
ModelConfigBase, ModelConfigBase,
ModelType, ModelType,
SubModelType, SubModelType,
calc_model_size_by_fs,
classproperty, classproperty,
) )

3
invokeai/backend/stable_diffusion/diffusers_pipeline.py
View File

@ -26,10 +26,9 @@ from pydantic import Field
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
from invokeai.app.services.config import InvokeAIAppConfig from invokeai.app.services.config import InvokeAIAppConfig
from invokeai.backend.ip_adapter.ip_adapter import IPAdapter, IPAdapterPlus from invokeai.backend.ip_adapter.ip_adapter import IPAdapter
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import ( from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
ConditioningData, ConditioningData,
IPAdapterConditioningInfo,
) )
from ..util import auto_detect_slice_size, normalize_device from ..util import auto_detect_slice_size, normalize_device