diffusers
Add LoRA support for Cosmos Predict 2.5 and fix pipeline to match official Cosmos repo
#13664
Merged
Merged
Add LoRA support for Cosmos Predict 2.5 and fix pipeline to match official Cosmos repo #13664
yiyixuxu
merged
19 commits into
huggingface:main
from
terarachang:cosmos_predict_2.5_lora_clean
support lora for cosmos 2.5
48fbe845
Fix inconsistencies with cosmos official repo in VAE encoding, text e…
52168162
Support f_min and f_max in linear_scheduler warmup
3d5551bc
Add requirements and dataset preprocessing scripts to run examples
f0b51ed3
Add LoRA training scripts
9038e10f
Add LoRA eval scripts
cee46cc8
add assets for blogpost
9e24319d
Fix(scheduler): device mismatch from upstream b114620 - move rk and b…
c3bb1aab
Always upcast to fp32
d32b085a
Directly inhrit from LoraBaseMixin
80aa547a
yiyixuxu
commented
on 2026-05-04
yiyixuxu
commented
on 2026-05-04
yiyixuxu
commented
on 2026-05-04
yiyixuxu
commented
on 2026-05-04
yiyixuxu
commented
on 2026-05-04
remove flash-attn2
c8513c20
Use _keep_in_fp32_modules instead of autocast
49f5b350
remove the get_latent_shape_cthw method and fix style
6acdc1c5
simplifiy the eval script to make it more user-friendly
d4b27f6a
yiyixuxu
commented
on 2026-05-05
yiyixuxu
approved these changes
on 2026-05-05
overwrite scheduling_unipc_multistep.py with main's version
3cc1210d
remove network_alphas and add # Copied from
38844fbc
remove figures and assets
119822fd
revert scheduler
770abee9
revert fp32 upcast and support bs > 1
13cfbb03
Login to write a write a comment.
Login via GitHub