Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,15 +60,22 @@ pipe = WanImageToVideoPipeline.from_pretrained(
|
|
| 60 |
# =========================================================
|
| 61 |
# LOAD LORA ADAPTERS
|
| 62 |
# =========================================================
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
pipe.load_lora_weights(
|
| 64 |
"lightx2v/Wan2.2-Lightning",
|
| 65 |
weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/high_noise_model.safetensors",
|
| 66 |
adapter_name="lightx2v"
|
| 67 |
)
|
|
|
|
| 68 |
pipe.load_lora_weights(
|
| 69 |
"obsxrver/wan2.2-i2v-scat",
|
| 70 |
-
weight_name="WAN2.2-I2V-
|
| 71 |
-
adapter_name="
|
|
|
|
| 72 |
)
|
| 73 |
pipe.load_lora_weights(
|
| 74 |
"lightx2v/Wan2.2-Lightning",
|
|
@@ -76,11 +83,7 @@ pipe.load_lora_weights(
|
|
| 76 |
adapter_name="lightx2v_2",
|
| 77 |
load_into_transformer_2=True
|
| 78 |
)
|
| 79 |
-
|
| 80 |
-
"obsxrver/wan2.2-i2v-scat",
|
| 81 |
-
weight_name="WAN2.2-I2V-LowNoise_scat-xxi-i2v.safetensors",
|
| 82 |
-
adapter_name="i2v_scat_2"
|
| 83 |
-
)
|
| 84 |
|
| 85 |
pipe.set_adapters([ "i2v_scat","lightx2v","i2v_scat_2","lightx2v_2"], adapter_weights=[1., 1., 1., 1.])
|
| 86 |
|
|
@@ -94,12 +97,12 @@ pipe.unload_lora_weights()
|
|
| 94 |
# =========================================================
|
| 95 |
# QUANTIZATION & AOT OPTIMIZATION
|
| 96 |
# =========================================================
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
|
| 100 |
|
| 101 |
-
|
| 102 |
-
|
| 103 |
|
| 104 |
# =========================================================
|
| 105 |
# DEFAULT PROMPTS
|
|
|
|
| 60 |
# =========================================================
|
| 61 |
# LOAD LORA ADAPTERS
|
| 62 |
# =========================================================
|
| 63 |
+
pipe.load_lora_weights(
|
| 64 |
+
"obsxrver/wan2.2-i2v-scat",
|
| 65 |
+
weight_name="WAN2.2-I2V-HighNoise_scat-xxi-i2v.safetensors",
|
| 66 |
+
adapter_name="i2v_scat"
|
| 67 |
+
)
|
| 68 |
pipe.load_lora_weights(
|
| 69 |
"lightx2v/Wan2.2-Lightning",
|
| 70 |
weight_name="Wan2.2-I2V-A14B-4steps-lora-rank64-Seko-V1/high_noise_model.safetensors",
|
| 71 |
adapter_name="lightx2v"
|
| 72 |
)
|
| 73 |
+
|
| 74 |
pipe.load_lora_weights(
|
| 75 |
"obsxrver/wan2.2-i2v-scat",
|
| 76 |
+
weight_name="WAN2.2-I2V-LowNoise_scat-xxi-i2v.safetensors",
|
| 77 |
+
adapter_name="i2v_scat_2",
|
| 78 |
+
load_into_transformer_2=True
|
| 79 |
)
|
| 80 |
pipe.load_lora_weights(
|
| 81 |
"lightx2v/Wan2.2-Lightning",
|
|
|
|
| 83 |
adapter_name="lightx2v_2",
|
| 84 |
load_into_transformer_2=True
|
| 85 |
)
|
| 86 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
|
| 88 |
pipe.set_adapters([ "i2v_scat","lightx2v","i2v_scat_2","lightx2v_2"], adapter_weights=[1., 1., 1., 1.])
|
| 89 |
|
|
|
|
| 97 |
# =========================================================
|
| 98 |
# QUANTIZATION & AOT OPTIMIZATION
|
| 99 |
# =========================================================
|
| 100 |
+
quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
|
| 101 |
+
quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
|
| 102 |
+
quantize_(pipe.transformer_2, Float8DynamicActivationFloat8WeightConfig())
|
| 103 |
|
| 104 |
+
aoti.aoti_blocks_load(pipe.transformer, 'zerogpu-aoti/Wan2', variant='fp8da')
|
| 105 |
+
aoti.aoti_blocks_load(pipe.transformer_2, 'zerogpu-aoti/Wan2', variant='fp8da')
|
| 106 |
|
| 107 |
# =========================================================
|
| 108 |
# DEFAULT PROMPTS
|