meepmoo commited on
Commit
61a8a3a
·
verified ·
1 Parent(s): 93af8bd

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +2 -2
worker_runpod.py CHANGED
@@ -93,8 +93,8 @@ with torch.inference_mode():
93
  }
94
 
95
  scheduler = sampler_dict["DPM++"].from_pretrained(model_id, subfolder="scheduler")
96
- lora_weight = values.get("lora_weight", 1.00)
97
- pipeline = merge_lora(pipeline, lora_path, lora_weight)
98
 
99
  if transformer.config.in_channels != vae.config.latent_channels:
100
  pipeline = CogVideoX_Fun_Pipeline_Inpaint.from_pretrained(
 
93
  }
94
 
95
  scheduler = sampler_dict["DPM++"].from_pretrained(model_id, subfolder="scheduler")
96
+
97
+ pipeline = merge_lora(pipeline, lora_path, 1.00)
98
 
99
  if transformer.config.in_channels != vae.config.latent_channels:
100
  pipeline = CogVideoX_Fun_Pipeline_Inpaint.from_pretrained(