support ltx2 two stage pipeline & vram

2026-03-18 22:08:13 +00:00 · 2026-01-30 16:55:40 +08:00
parent b1a2782ad7
commit 4f23caa55f
7 changed files with 227 additions and 85 deletions
--- a/diffsynth/utils/data/media_io.py
+++ b/diffsynth/utils/data/media_io.py
@@ -3,6 +3,7 @@ from fractions import Fraction
 import torch
 import av
 from tqdm import tqdm
+from PIL import Image


 def _resample_audio(