Test multi

usnistgov · Nov 28, 2024 · 42e6190 · 42e6190
1 parent 1d00be9
commit 42e6190
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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