rahul7star commited on
Commit
d82d0e1
·
verified ·
1 Parent(s): 02c2928

Update optimization.py

Browse files
Files changed (1) hide show
  1. optimization.py +2 -2
optimization.py CHANGED
@@ -40,8 +40,8 @@ INDUCTOR_CONFIGS = {
40
 
41
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
42
 
43
- @spaces.GPU(duration=1500)
44
- def compile_transformer():
45
  # --- LoRA 1: lightning (loads into default transformer) ---
46
  pipeline.load_lora_weights(
47
  "Kijai/WanVideo_comfy",
 
40
 
41
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
42
 
43
+ @spaces.GPU(duration=1500)
44
+ def compile_transformer():
45
  # --- LoRA 1: lightning (loads into default transformer) ---
46
  pipeline.load_lora_weights(
47
  "Kijai/WanVideo_comfy",