mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-04-30 03:01:28 +00:00
Removed all submodules. Submodule free now, yay.
This commit is contained in:
@@ -7,9 +7,6 @@ import weakref
|
||||
from typing import Union, TYPE_CHECKING
|
||||
|
||||
from diffusers import Transformer2DModel
|
||||
from transformers import T5EncoderModel, CLIPTextModel, CLIPTokenizer, T5Tokenizer, CLIPVisionModelWithProjection
|
||||
from toolkit.paths import REPOS_ROOT
|
||||
sys.path.append(REPOS_ROOT)
|
||||
|
||||
|
||||
if TYPE_CHECKING:
|
||||
|
||||
@@ -8,7 +8,6 @@ from toolkit.config_modules import GenerateImageConfig, ModelConfig
|
||||
from toolkit.dequantize import patch_dequantization_on_save
|
||||
from toolkit.models.base_model import BaseModel
|
||||
from toolkit.prompt_utils import PromptEmbeds
|
||||
from toolkit.paths import REPOS_ROOT
|
||||
from transformers import AutoTokenizer, UMT5EncoderModel
|
||||
from diffusers import AutoencoderKLWan, WanPipeline, WanTransformer3DModel
|
||||
import os
|
||||
@@ -34,7 +33,6 @@ from diffusers import FlowMatchEulerDiscreteScheduler, UniPCMultistepScheduler
|
||||
from typing import TYPE_CHECKING, List
|
||||
from toolkit.accelerator import unwrap_model
|
||||
from toolkit.samplers.custom_flowmatch_sampler import CustomFlowMatchEulerDiscreteScheduler
|
||||
from torchvision.transforms import Resize, ToPILImage
|
||||
from tqdm import tqdm
|
||||
import torch.nn.functional as F
|
||||
from diffusers.pipelines.wan.pipeline_output import WanPipelineOutput
|
||||
|
||||
@@ -6,7 +6,6 @@ from toolkit.accelerator import unwrap_model
|
||||
from toolkit.basic import flush
|
||||
from toolkit.config_modules import GenerateImageConfig, ModelConfig
|
||||
from toolkit.prompt_utils import PromptEmbeds
|
||||
from toolkit.paths import REPOS_ROOT
|
||||
from transformers import AutoTokenizer, UMT5EncoderModel
|
||||
from diffusers import AutoencoderKLWan, WanImageToVideoPipeline, WanTransformer3DModel
|
||||
import os
|
||||
|
||||
Reference in New Issue
Block a user