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

Full test of ui arch compatability

Browse Source
This commit is contained in:
Jaret Burkett
2026-08-28 06:56:05 -06:00
parent 520d96aac3
commit c3bc8b0b4e
4 changed files with 105 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
toolkit/models/v2/diffusion_models/wan.py
View File

@@ -30,6 +30,19 @@ class WanTransformer3DModel(DiffusersWanTransformer3DModel, OstrisModelMixin):
("Wan-AI/Wan2.2-I2V-A14B-Diffusers", "transformer_2"): [
"split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors",
],
# the UI defaults point at the ai-toolkit bf16 repacks; same comfy files
("ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16", "transformer"): [
"split_files/diffusion_models/wan2.2_t2v_high_noise_14B_fp8_scaled.safetensors",
],
("ai-toolkit/Wan2.2-T2V-A14B-Diffusers-bf16", "transformer_2"): [
"split_files/diffusion_models/wan2.2_t2v_low_noise_14B_fp8_scaled.safetensors",
],
("ai-toolkit/Wan2.2-I2V-A14B-Diffusers-bf16", "transformer"): [
"split_files/diffusion_models/wan2.2_i2v_high_noise_14B_fp8_scaled.safetensors",
],
("ai-toolkit/Wan2.2-I2V-A14B-Diffusers-bf16", "transformer_2"): [
"split_files/diffusion_models/wan2.2_i2v_low_noise_14B_fp8_scaled.safetensors",
],
"Wan-AI/Wan2.1-T2V-1.3B-Diffusers": {
"repo": "Comfy-Org/Wan_2.1_ComfyUI_repackaged",
"files": [