From ef1d60ba3409dfb638bec539ec824ec34e11b0ea Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Wed, 30 Jul 2025 10:13:22 -0600
Subject: [PATCH] Update wan 2.2 5b timestep distribution to weighted.

---
 ui/src/app/jobs/new/options.ts | 1 +
 version.py                     | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/ui/src/app/jobs/new/options.ts b/ui/src/app/jobs/new/options.ts
index 6fcfc8f6..03ed453f 100644
--- a/ui/src/app/jobs/new/options.ts
+++ b/ui/src/app/jobs/new/options.ts
@@ -198,6 +198,7 @@ export const modelArchs: ModelArch[] = [
       'config.process[0].sample.fps': [24, 1],
       'config.process[0].sample.width': [768, 1024],
       'config.process[0].sample.height': [768, 1024],
+      'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
     },
     disableSections: ['network.conv'],
     additionalSections: ['sample.ctrl_img', 'datasets.num_frames', 'model.low_vram'],
diff --git a/version.py b/version.py
index 077df00b..814e9f56 100644
--- a/version.py
+++ b/version.py
@@ -1 +1 @@
-VERSION = "0.3.15"
\ No newline at end of file
+VERSION = "0.3.16"
\ No newline at end of file