From 42e6190abb05fb08a7736be9afcb6951d3fa87db Mon Sep 17 00:00:00 2001
From: knc6 <writetokamal.1989@gmail.com>
Date: Wed, 27 Nov 2024 20:46:00 -0500
Subject: [PATCH] Test multi

---
 atomgpt/inverse_models/inverse_models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/atomgpt/inverse_models/inverse_models.py b/atomgpt/inverse_models/inverse_models.py
index 0677c99..e3c5852 100644
--- a/atomgpt/inverse_models/inverse_models.py
+++ b/atomgpt/inverse_models/inverse_models.py
@@ -244,7 +244,7 @@ def batch_evaluate(
     for batch_start in tqdm(range(0, len(prompts), batch_size)):
         batch_end = min(batch_start + batch_size, len(prompts))
         batch_prompts = prompts[batch_start:batch_end]
-
+        # print("batch_prompts",batch_prompts)
         # Tokenize and prepare inputs
         inputs = tokenizer(
             [
@@ -254,6 +254,7 @@ def batch_evaluate(
             return_tensors="pt",
             padding=True,
             truncation=True,
+            max_length=config.max_seq_length,
         ).to("cuda")
 
         # Generate outputs using the model