This commit is contained in:
SolitaryThinker
2025-09-12 07:23:54 +00:00
parent 93ebd15a0d
commit dc9a1b9ec8
38 changed files with 8782 additions and 12 deletions
@@ -16,6 +16,7 @@ def main():
use_fsdp_inference=True,
text_encoder_cpu_offload=False,
dit_cpu_offload=False,
use_sf_wan=True,
)
sampling_param = SamplingParam.from_pretrained(model_name)