Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- optimization.py +4 -15
optimization.py
CHANGED
@@ -64,7 +64,7 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
64 |
# pipeline.unload_lora_weights()
|
65 |
|
66 |
|
67 |
-
|
68 |
pipeline.load_lora_weights(
|
69 |
"Kijai/WanVideo_comfy",
|
70 |
weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors",
|
@@ -72,23 +72,12 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
72 |
)
|
73 |
kwargs_lora = {}
|
74 |
kwargs_lora["load_into_transformer_2"] = True
|
75 |
-
#Low
|
76 |
pipeline.load_lora_weights(
|
77 |
-
"
|
78 |
-
|
|
|
79 |
adapter_name="lightning_2", **kwargs_lora
|
80 |
)
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
pipeline.set_adapters(["lightning", "lightning_2"], adapter_weights=[1., 1.])
|
93 |
|
94 |
|
|
|
64 |
# pipeline.unload_lora_weights()
|
65 |
|
66 |
|
67 |
+
|
68 |
pipeline.load_lora_weights(
|
69 |
"Kijai/WanVideo_comfy",
|
70 |
weight_name="Lightx2v/lightx2v_T2V_14B_cfg_step_distill_v2_lora_rank128_bf16.safetensors",
|
|
|
72 |
)
|
73 |
kwargs_lora = {}
|
74 |
kwargs_lora["load_into_transformer_2"] = True
|
|
|
75 |
pipeline.load_lora_weights(
|
76 |
+
"TheRaf7/instagirl-v2",
|
77 |
+
weight_name="Instagirlv2.0_lownoise.safetensors",
|
78 |
+
#weight_name="Wan22-Lightning/Wan2.2-Lightning_T2V-A14B-4steps-lora_LOW_fp16.safetensors",
|
79 |
adapter_name="lightning_2", **kwargs_lora
|
80 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
81 |
pipeline.set_adapters(["lightning", "lightning_2"], adapter_weights=[1., 1.])
|
82 |
|
83 |
|