From 303eb29eced587ffce16e2f204787c17e4927404 Mon Sep 17 00:00:00 2001
From: aiXander <xandersteenbruggex@gmail.com>
Date: Mon, 8 Apr 2024 01:30:16 -0700
Subject: [PATCH] fix tiny bug in predict

---
 predict.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/predict.py b/predict.py
index 5eaef57..d0f1baa 100755
--- a/predict.py
+++ b/predict.py
@@ -264,7 +264,7 @@ class Predictor(BasePredictor):
         attributes['grid_prompts'] = config.training_attributes["validation_prompts"]
         attributes['job_time_seconds'] = config.job_time
 
-        print(f"LORA training finished in {runtime:.1f} seconds")
+        print(f"LORA training finished in {config.job_time:.1f} seconds")
         print(f"Returning {out_path}")
 
         if DEBUG_MODE or debug: