Fixed assignment error in AssignmentInstructions.parse_assign_line()

Author: CrudeDiatribe

backends/model_converter/no_pickle_fake_torch.py

@@ -27,8 +27,10 @@ def extract_weights_from_checkpoint(fb0):
         with myzip.open(folder_name + f'/data/{load_instruction.obj_key}') as myfile:
           if (load_instruction.load_from_file_buffer(myfile)):
             torch_weights['state_dict'][sd_key] = load_instruction.get_data()
-      for sd_key,special in special_instructions.items():
-        torch_weights['state_dict'][sd_key] = special
+      if len(special_instructions) > 0:
+        torch_weights['state_dict']['_metadata'] = {}
+        for sd_key,special in special_instructions.items():
+          torch_weights['state_dict']['_metadata'][sd_key] = special
   return torch_weights
 
 def examine_pickle(fb0):
@@ -96,7 +98,7 @@ def parse_assign_line(self, line):
     del huge_mess
     assignments[-1] = assignments[-1].strip('}')
     re_var = re.compile('^_var\d+$')
-    assignment_count
+    assignment_count = 0
     for a in assignments:
       if self._add_assignment(a, re_var):
         assignment_count = assignment_count + 1