shorter import paths

This commit is contained in:
Pierre Chapuis 2023-08-28 16:29:38 +02:00
parent 8615dbdbde
commit 18c84c7b72
8 changed files with 24 additions and 23 deletions

View file

@ -6,9 +6,11 @@ from torch import nn
from diffusers import ControlNetModel # type: ignore from diffusers import ControlNetModel # type: ignore
from refiners.fluxion.utils import save_to_safetensors from refiners.fluxion.utils import save_to_safetensors
from refiners.fluxion.model_converter import ModelConverter from refiners.fluxion.model_converter import ModelConverter
from refiners.foundationals.latent_diffusion.stable_diffusion_1 import SD1Controlnet from refiners.foundationals.latent_diffusion import (
from refiners.foundationals.latent_diffusion.schedulers.dpm_solver import DPMSolver SD1UNet,
from refiners.foundationals.latent_diffusion import SD1UNet SD1Controlnet,
DPMSolver,
)
class Args(argparse.Namespace): class Args(argparse.Namespace):

View file

@ -9,7 +9,7 @@ from diffusers import DiffusionPipeline # type: ignore
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl
from refiners.fluxion.model_converter import ModelConverter from refiners.fluxion.model_converter import ModelConverter
from refiners.fluxion.utils import save_to_safetensors from refiners.fluxion.utils import save_to_safetensors
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import SD1UNet from refiners.foundationals.latent_diffusion import SD1UNet
from refiners.foundationals.latent_diffusion.lora import LoraTarget, apply_loras_to_target from refiners.foundationals.latent_diffusion.lora import LoraTarget, apply_loras_to_target
from refiners.adapters.lora import Lora from refiners.adapters.lora import Lora

View file

@ -12,8 +12,8 @@ from convert_transformers_clip_text_model import (
Args as TextEncoderConversionArgs, Args as TextEncoderConversionArgs,
) )
from refiners.foundationals.clip.text_encoder import CLIPTextEncoderL from refiners.foundationals.clip.text_encoder import CLIPTextEncoderL
from refiners.foundationals.latent_diffusion import SD1UNet
from refiners.foundationals.latent_diffusion.lora import LoraTarget from refiners.foundationals.latent_diffusion.lora import LoraTarget
from refiners.foundationals.latent_diffusion.stable_diffusion_1 import SD1UNet
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl

View file

@ -5,20 +5,14 @@ from refiners.foundationals.clip.text_encoder import (
CLIPTextEncoderL, CLIPTextEncoderL,
) )
from refiners.foundationals.latent_diffusion.schedulers import Scheduler, DPMSolver from refiners.foundationals.latent_diffusion.schedulers import Scheduler, DPMSolver
from refiners.foundationals.latent_diffusion.stable_diffusion_1.model import ( from refiners.foundationals.latent_diffusion.stable_diffusion_1 import (
StableDiffusion_1, StableDiffusion_1,
StableDiffusion_1_Inpainting, StableDiffusion_1_Inpainting,
)
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import (
SD1UNet, SD1UNet,
)
from refiners.foundationals.latent_diffusion.stable_diffusion_1.controlnet import (
SD1Controlnet, SD1Controlnet,
) )
from refiners.foundationals.latent_diffusion.stable_diffusion_xl.unet import ( from refiners.foundationals.latent_diffusion.stable_diffusion_xl import (
SDXLUNet, SDXLUNet,
)
from refiners.foundationals.latent_diffusion.stable_diffusion_xl.text_encoder import (
DoubleTextEncoder, DoubleTextEncoder,
) )

View file

@ -10,7 +10,7 @@ from refiners.fluxion.layers import (
Parallel, Parallel,
) )
from refiners.adapters.adapter import Adapter from refiners.adapters.adapter import Adapter
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import SD1UNet from refiners.foundationals.latent_diffusion import SD1UNet
from refiners.foundationals.latent_diffusion.cross_attention import CrossAttentionBlock from refiners.foundationals.latent_diffusion.cross_attention import CrossAttentionBlock
from torch import Tensor from torch import Tensor

View file

@ -1,20 +1,22 @@
from dataclasses import dataclass from dataclasses import dataclass
from typing import Any, TypeVar, TypedDict, cast from typing import Any, TypeVar, TypedDict, cast
from pydantic import BaseModel from pydantic import BaseModel
from refiners.foundationals.latent_diffusion.schedulers.ddpm import DDPM
from torch import device as Device, Tensor, randn, dtype as DType, Generator, cat from torch import device as Device, Tensor, randn, dtype as DType, Generator, cat
from loguru import logger from loguru import logger
from torch.utils.data import Dataset from torch.utils.data import Dataset
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import SD1UNet
from refiners.foundationals.clip.text_encoder import CLIPTextEncoderL from refiners.foundationals.clip.text_encoder import CLIPTextEncoderL
from refiners.foundationals.latent_diffusion.auto_encoder import LatentDiffusionAutoencoder
from torchvision.transforms import RandomCrop # type: ignore from torchvision.transforms import RandomCrop # type: ignore
import refiners.fluxion.layers as fl import refiners.fluxion.layers as fl
from PIL import Image from PIL import Image
from functools import cached_property from functools import cached_property
from refiners.training_utils.config import BaseConfig from refiners.training_utils.config import BaseConfig
from refiners.foundationals.latent_diffusion import StableDiffusion_1 from refiners.foundationals.latent_diffusion import (
from refiners.foundationals.latent_diffusion.schedulers import DPMSolver StableDiffusion_1,
DPMSolver,
SD1UNet,
LatentDiffusionAutoencoder,
)
from refiners.foundationals.latent_diffusion.schedulers import DDPM
from torch.nn.functional import mse_loss from torch.nn.functional import mse_loss
import random import random
from refiners.training_utils.wandb import WandbLoggable from refiners.training_utils.wandb import WandbLoggable

View file

@ -8,9 +8,12 @@ from PIL import Image
from pathlib import Path from pathlib import Path
from refiners.fluxion.utils import load_from_safetensors, image_to_tensor, manual_seed from refiners.fluxion.utils import load_from_safetensors, image_to_tensor, manual_seed
from refiners.foundationals.latent_diffusion import StableDiffusion_1, StableDiffusion_1_Inpainting from refiners.foundationals.latent_diffusion import (
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import SD1UNet StableDiffusion_1,
from refiners.foundationals.latent_diffusion.stable_diffusion_1.controlnet import SD1Controlnet StableDiffusion_1_Inpainting,
SD1UNet,
SD1Controlnet,
)
from refiners.foundationals.latent_diffusion.lora import LoraWeights from refiners.foundationals.latent_diffusion.lora import LoraWeights
from refiners.foundationals.latent_diffusion.schedulers import DDIM from refiners.foundationals.latent_diffusion.schedulers import DDIM
from refiners.foundationals.latent_diffusion.self_attention_injection import SelfAttentionInjection from refiners.foundationals.latent_diffusion.self_attention_injection import SelfAttentionInjection

View file

@ -1,4 +1,4 @@
from refiners.foundationals.latent_diffusion.stable_diffusion_1.unet import SD1UNet from refiners.foundationals.latent_diffusion import SD1UNet
from refiners.fluxion import manual_seed from refiners.fluxion import manual_seed
import torch import torch