theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-20 07:18:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

update model loader

Browse Source
This commit is contained in:
Artiprocher
2024-10-08 16:46:44 +08:00
parent 892f80c265
commit ec352cfce2
4 changed files with 189 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/model_manager.py
View File

@@ -4,7 +4,7 @@ from torch import Tensor
from typing_extensions import Literal, TypeAlias
from typing import List
from .downloader import download_models, Preset_model_id, Preset_model_website
from .downloader import download_models, download_customized_models, Preset_model_id, Preset_model_website
from .sd_text_encoder import SDTextEncoder
from .sd_unet import SDUNet