File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 32
32
def merge (args ):
33
33
model_dict = {}
34
34
# load the first item: vision_model
35
- visual_encoer_state_dict = paddle .load (args .vision_name_or_path )
36
- for n , p in visual_encoer_state_dict .items ():
35
+ visual_encoder_state_dict = paddle .load (args .vision_name_or_path )
36
+ for n , p in visual_encoder_state_dict .items ():
37
37
if n .startswith ("visual_encoder" ):
38
38
model_dict [n ] = p
39
39
else :
40
- model_dict ["visual_encoer ." + n ] = p
40
+ model_dict ["visual_encoder ." + n ] = p
41
41
logger .info ("[1/2] load visual_encoder done!" )
42
42
# load the second item: Qformer
43
- visual_encoer_state_dict = paddle .load (args .vision_name_or_path )
44
- for n , p in visual_encoer_state_dict .items ():
43
+ visual_encoder_state_dict = paddle .load (args .vision_name_or_path )
44
+ for n , p in visual_encoder_state_dict .items ():
45
45
if n .startswith ("Qformer" ):
46
46
model_dict [n ] = p
47
47
else :
You can’t perform that action at this time.
0 commit comments