File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -4810,12 +4810,11 @@ def _maybe_expand_t2v_lora_for_i2v(
48104810 transformer : torch .nn .Module ,
48114811 state_dict ,
48124812 ):
4813- if transformer .config .image_dim is None :
4814- return state_dict
4813+ print ("wtf 0" , hasattr (transformer , 'vace_blocks' ))
4814+ # if transformer.config.image_dim is None:
4815+ # return state_dict
48154816
48164817 target_device = transformer .device
4817- print ("expanding t2v to i2v" )
4818- print ("wtf 0" ,hasattr (transformer , 'vace_blocks' ))
48194818 if any (k .startswith ("transformer.blocks." ) for k in state_dict ):
48204819 num_blocks = len ({k .split ("blocks." )[1 ].split ("." )[0 ] for k in state_dict if "blocks." in k })
48214820 is_i2v_lora = any ("add_k_proj" in k for k in state_dict ) and any ("add_v_proj" in k for k in state_dict )
You can’t perform that action at this time.
0 commit comments