fix

2026-01-05 08:30:16 +00:00
parent 4ce4ac4734
commit 14e334d0b4
3 changed files with 116 additions and 57 deletions
@@ -21,6 +21,7 @@ def main():
        dit_cpu_offload=False,
        vae_cpu_offload=False,
        VSA_sparsity=0.8,
+        init_weights_from_safetensors="/mnt/weka/home/hao.zhang/wl/release/dmd_distill_1.3_4n_syn/checkpoint-900_weight_only/generator_inference_transformer"
    )
    load_end_time = time.perf_counter()
    load_time = load_end_time - load_start_time