ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added initial support for finetuning wan i2v WIP

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-07 20:34:38 -06:00
parent 38ad5a4644
commit a8680c75eb
10 changed files with 575 additions and 286 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
toolkit/config_modules.py
View File

@@ -540,6 +540,11 @@ class ModelConfig:
self.arch: ModelArch = kwargs.get("arch", None)
# can be used to load the extras like text encoder or vae from here
# only setup for some models but will prevent having to download the te for
# 20 different model variants
self.extras_name_or_path = kwargs.get("extras_name_or_path", self.name_or_path)
# kwargs to pass to the model
self.model_kwargs = kwargs.get("model_kwargs", {})