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

Added 4bit ARAs for Wan 2.2 14b models

Browse Source
This commit is contained in:
Jaret Burkett
2025-08-21 08:16:07 -06:00
parent 83deaec417
commit 38d3814be7
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ui/src/app/jobs/new/options.ts
View File

@@ -214,9 +214,10 @@ export const modelArchs: ModelArch[] = [
}, },
disableSections: ['network.conv'], disableSections: ['network.conv'],
additionalSections: ['datasets.num_frames', 'model.low_vram', 'model.multistage'], additionalSections: ['datasets.num_frames', 'model.low_vram', 'model.multistage'],
// accuracyRecoveryAdapters: { accuracyRecoveryAdapters: {
// '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint3.safetensors', // '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint3.safetensors',
// }, '4 bit with ARA': 'uint4|ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint4.safetensors',
},
}, },
{ {
name: 'wan22_14b_i2v', name: 'wan22_14b_i2v',
@@ -244,6 +245,9 @@ export const modelArchs: ModelArch[] = [
}, },
disableSections: ['network.conv'], disableSections: ['network.conv'],
additionalSections: ['sample.ctrl_img', 'datasets.num_frames', 'model.low_vram', 'model.multistage'], additionalSections: ['sample.ctrl_img', 'datasets.num_frames', 'model.low_vram', 'model.multistage'],
accuracyRecoveryAdapters: {
'4 bit with ARA': 'uint4|ostris/accuracy_recovery_adapters/wan22_14b_i2v_torchao_uint4.safetensors',
},
}, },
{ {
name: 'wan22_5b', name: 'wan22_5b',

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.5.2" VERSION = "0.5.3"