From 38d3814be7a4fed56f748924dc891c2499cdeb22 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Thu, 21 Aug 2025 08:16:07 -0600
Subject: [PATCH] Added 4bit ARAs for Wan 2.2 14b models

---
 ui/src/app/jobs/new/options.ts | 10 +++++++---
 version.py                     |  2 +-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index c180c240..1381495d 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -214,9 +214,10 @@ export const modelArchs: ModelArch[] = [
     },
     disableSections: ['network.conv'],
     additionalSections: ['datasets.num_frames', 'model.low_vram', 'model.multistage'],
-    // accuracyRecoveryAdapters: {
-    //   '3 bit with ARA': 'uint3|ostris/accuracy_recovery_adapters/wan22_14b_t2i_torchao_uint3.safetensors',
-    // },
+    accuracyRecoveryAdapters: {
+      // '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',
@@ -244,6 +245,9 @@ export const modelArchs: ModelArch[] = [
     },
     disableSections: ['network.conv'],
     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',
diff --git a/version.py b/version.py
index 214abb41..a69dd004 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.5.2"
\ No newline at end of file
+VERSION = "0.5.3"
\ No newline at end of file