github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Added initial support for training i2v adapter WIP

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-09 08:06:29 -06:00
parent a8680c75eb
commit 615b0d0e94
7 changed files with 682 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/sampler.py
View File

@@ -164,11 +164,12 @@ def get_sampler(
config_to_use = copy.deepcopy(flux_config)
if arch == "sd":
config_to_use = copy.deepcopy(sd_flow_config)
if arch == "flux":
elif arch == "flux":
config_to_use = copy.deepcopy(flux_config)
elif arch == "lumina2":
config_to_use = copy.deepcopy(lumina2_config)
else:
print(f"Unknown architecture {arch}, using default flux config")
# use flux by default
config_to_use = copy.deepcopy(flux_config)
else: