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):
8080 # _var2262 = {'model.diffusion_model.input_blocks.0.0.weight': _var1, 'model.diffusion_model.input_blocks.0.0.bias': _var3,\
8181 # ...\
8282 # '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 )
8484 assignments = huge_mess .split (', ' )
8585 del huge_mess
8686 assignments [- 1 ] = assignments [- 1 ].strip ('}' )
@@ -105,7 +105,7 @@ def parse_update_line(self, line):
105105 # 'cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.weight': _var2003,\
106106 # ...\
107107 #'cond_stage_model.transformer.text_model.final_layer_norm.bias': _var2261})
108- garbage , huge_mess = line .split ('({' )
108+ garbage , huge_mess = line .split ('({' , 1 )
109109 updates = huge_mess .split (', ' )
110110 del huge_mess
111111 updates [- 1 ] = updates [- 1 ].strip ('})' )
You can’t perform that action at this time.
0 commit comments