obsxrver commited on
Commit
f94dc76
·
verified ·
1 Parent(s): 1835cde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -12
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-HighNoise_scat-xxi-i2v.safetensors",
71
- adapter_name="i2v_scat"
 
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
- pipe.load_lora_weights(
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
- #quantize_(pipe.text_encoder, Int8WeightOnlyConfig())
98
- #quantize_(pipe.transformer, Float8DynamicActivationFloat8WeightConfig())
99
- #quantize_(pipe.transformer_2, Float8DynamicActivationFloat8WeightConfig())
100
 
101
- #aoti.aoti_blocks_load(pipe.transformer, 'zerogpu-aoti/Wan2', variant='bf16')
102
- #aoti.aoti_blocks_load(pipe.transformer_2, 'zerogpu-aoti/Wan2', variant='bf16')
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