Skip to content

Commit 5c9d460

Browse files
committed
remove double loading the model.
1 parent 77a43cf commit 5c9d460

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

src/maxdiffusion/generate_wan.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -139,9 +139,6 @@ def run(config, pipeline=None, filename_prefix=""):
139139

140140
checkpoint_loader = WanCheckpointer(config, "WAN_CHECKPOINT")
141141
pipeline, _, _ = checkpoint_loader.load_checkpoint()
142-
pipeline_lib = get_pipeline(model_key)
143-
WanPipeline = pipeline_lib.WanPipeline
144-
pipeline = WanPipeline.from_pretrained(config)
145142
s0 = time.perf_counter()
146143

147144
# Using global_batch_size_to_train_on so not to create more config variables

0 commit comments

Comments
 (0)