all files migrated; tweaks needed

This commit is contained in:
Lincoln Stein
2023-03-03 00:02:15 -05:00
parent 3f0b0f3250
commit 6a990565ff
496 changed files with 276 additions and 934 deletions

View File

@ -10,7 +10,7 @@ import traceback
from typing import Callable
from urllib import request, error as ul_error
from huggingface_hub import HfFolder, hf_hub_url, ModelSearchArguments, ModelFilter, HfApi
from ldm.invoke.globals import Globals
from invokeai.backend.globals import Globals
class HuggingFaceConceptsLibrary(object):
def __init__(self, root=None):

View File

@ -26,11 +26,11 @@ from torchvision.transforms.functional import resize as tv_resize
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
from typing_extensions import ParamSpec
from ldm.invoke.globals import Globals
from ..stable_diffusion.diffusion import InvokeAIDiffuserComponent, PostprocessingSettings, AttentionMapSaver
from ..stable_diffusion.textual_inversion_manager import TextualInversionManager
from ..stable_diffusion.offloading import LazilyLoadedModelGroup, FullyLoadedModelGroup, ModelGroup
from ..devices import normalize_device, CPU_DEVICE
from invokeai.backend.globals import Globals
from .diffusion import InvokeAIDiffuserComponent, PostprocessingSettings, AttentionMapSaver
from .textual_inversion_manager import TextualInversionManager
from .offloading import LazilyLoadedModelGroup, FullyLoadedModelGroup, ModelGroup
from ..util import normalize_device, CPU_DEVICE
from compel import EmbeddingsProvider
@dataclass

View File

@ -15,7 +15,7 @@ from torch import nn
from compel.cross_attention_control import Arguments
from diffusers.models.unet_2d_condition import UNet2DConditionModel
from diffusers.models.cross_attention import AttnProcessor
from ...devices import torch_dtype
from ...util import torch_dtype
class CrossAttentionType(enum.Enum):

View File

@ -23,7 +23,7 @@ from omegaconf import ListConfig
import urllib
from ..textual_inversion_manager import TextualInversionManager
from ...util import (
from ...util.util import (
log_txt_as_img,
exists,
default,

View File

@ -4,7 +4,7 @@ import torch
import numpy as np
from tqdm import tqdm
from functools import partial
from ...devices import choose_torch_device
from ...util import choose_torch_device
from .shared_invokeai_diffusion import InvokeAIDiffuserComponent
from .sampler import Sampler
from ..diffusionmodules.util import noise_like

View File

@ -7,7 +7,7 @@ import torch
import numpy as np
from tqdm import tqdm
from functools import partial
from ...devices import choose_torch_device
from ...util import choose_torch_device
from .shared_invokeai_diffusion import InvokeAIDiffuserComponent
from ..diffusionmodules.util import (

View File

@ -8,7 +8,7 @@ import torch
from diffusers.models.cross_attention import AttnProcessor
from typing_extensions import TypeAlias
from ldm.invoke.globals import Globals
from invokeai.backend.globals import Globals
from .cross_attention_control import Arguments, \
restore_default_cross_attention, override_cross_attention, Context, get_cross_attention_modules, \
CrossAttentionType, SwapCrossAttnContext

View File

@ -15,7 +15,7 @@ import torch.nn as nn
import numpy as np
from einops import repeat
from ...util import instantiate_from_config
from ...util.util import instantiate_from_config
def make_beta_schedule(

View File

@ -10,7 +10,7 @@ from einops import repeat
from transformers import CLIPTokenizer, CLIPTextModel
from ldm.invoke.devices import choose_torch_device
from ldm.invoke.globals import global_cache_dir
from invokeai.backend.globals import global_cache_dir
from ldm.modules.x_transformer import (
Encoder,
TransformerWrapper,