Update prompt_stretch for evo2

- Remove prompt_stretch option from run.sh
- Adjust condition in test_evo2_generation_batched.py to allow prompt stretching based on batch size

evo2/run.sh

@@ -14,7 +14,6 @@ EVO_CMD="numactl -m 1 -N 1 \
             --model_name ${MODEL_NAME} \
             --local_path ${MODEL_PATH} \
             --n_tokens 500 \
-            --prompt_stretch \
             --trace_gzip \
             --trace_logdir ./log/pt-trace/"
 

evo2/test_evo2_generation_batched.py

@@ -299,7 +299,7 @@ def main():
     print("[DEBUG] Prompt lengths:", [len(seq) for seq in sequences])
 
     # Debugging: replace all prompts with the longest prompt
-    if args.prompt_stretch:
+    if args.prompt_stretch or args.batch_size > 1:
         longest_prompt = max(sequences, key=len)
         sequences = [longest_prompt] * len(sequences)
         print(