File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -80,7 +80,7 @@ def parse_assign_line(self, line):
80
80
# _var2262 = {'model.diffusion_model.input_blocks.0.0.weight': _var1, 'model.diffusion_model.input_blocks.0.0.bias': _var3,\
81
81
# ...\
82
82
# 'cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm2.weight': _var1999}
83
- garbage , huge_mess = line .split (' = {' )
83
+ garbage , huge_mess = line .split (' = {' , 1 )
84
84
assignments = huge_mess .split (', ' )
85
85
del huge_mess
86
86
assignments [- 1 ] = assignments [- 1 ].strip ('}' )
@@ -105,7 +105,7 @@ def parse_update_line(self, line):
105
105
# 'cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.weight': _var2003,\
106
106
# ...\
107
107
#'cond_stage_model.transformer.text_model.final_layer_norm.bias': _var2261})
108
- garbage , huge_mess = line .split ('({' )
108
+ garbage , huge_mess = line .split ('({' , 1 )
109
109
updates = huge_mess .split (', ' )
110
110
del huge_mess
111
111
updates [- 1 ] = updates [- 1 ].strip ('})' )
You can’t perform that action at this time.
0 commit comments