Update modeling_GOT.py

modeling_GOT.py

@@ -575,17 +575,16 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                 stopping_criteria=[stopping_criteria]
                 )
             
-
+            outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
+            
+            if outputs.endswith(stop_str):
+                outputs = outputs[:-len(stop_str)]
+            outputs = outputs.strip()
+            
             if render:
                 print('==============rendering===============')
                 from .render_tools import svg_to_html, content_mmd_to_html, tik_html, translation_table
 
-                outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
-                
-                if outputs.endswith(stop_str):
-                    outputs = outputs[:-len(stop_str)]
-                outputs = outputs.strip()
-
                 if '**kern' in outputs:
                     import verovio
                     from cairosvg import svg2png
@@ -813,16 +812,16 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
                 max_new_tokens=4096,
                 stopping_criteria=[stopping_criteria]
                 )
-            
+
+        outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
+        
+        if outputs.endswith(stop_str):
+            outputs = outputs[:-len(stop_str)]
+        outputs = outputs.strip()   
+
         if render:
             print('==============rendering===============')
             from .render_tools import content_mmd_to_html
-            outputs = tokenizer.decode(output_ids[0, input_ids.shape[1]:]).strip()
-            
-            if outputs.endswith(stop_str):
-                outputs = outputs[:-len(stop_str)]
-            outputs = outputs.strip()
-
             html_path_2 = save_render_file
             right_num = outputs.count('\\right')
             left_num = outputs.count('\left')