Update handler.py

handler.py

@@ -42,6 +42,7 @@ class EndpointHandler:
             if k=='instruction':
                 prompt+=f'<{v}>' if isinstance(v, str) else "".join([f'<{v_i}>' for v_i in v])
             elif k=='up':
+                if v 
                 prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
             elif k=='down':
                 prompt+=f'{multi_modal_prefix}<{k}>{v}</{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
@@ -137,7 +138,7 @@ class EndpointHandler:
         
         # Process generated up/down lists
         processed_outputs = {"up": [], "down": []}
-        if mode == 'meta2diff':
+        if mode in ['meta2diff', 'meta2diff2compound']:
             for output in outputs:
                 up_split_index = output.index(self.tokenizer.convert_tokens_to_ids('</up>'))
                 generated_up_raw = [i.strip() for i in self.tokenizer.convert_ids_to_tokens(output[:up_split_index])]