Lengyue 1 год назад
Родитель
Сommit
1e6d715784
1 измененных файлов с 1 добавлено и 0 удалено
  1. 1 0
      tools/llama/generate.py

+ 1 - 0
tools/llama/generate.py

@@ -498,6 +498,7 @@ def generate_long(
 
 
         if use_prompt:
         if use_prompt:
             seg_idx = 1
             seg_idx = 1
+            global_encoded.append(encoded[0])
 
 
         while seg_idx < len(encoded):
         while seg_idx < len(encoded):
             logger.info(
             logger.info(