Skip to content

Commit 8b73e58

Browse files
committed
Put cap on initial split() in both AssignInstructions.parse* methods
1 parent 7d32f15 commit 8b73e58

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/model_converter/no_pickle_fake_torch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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('})')

0 commit comments

Comments
 (0)