fix tiny bug in predict

2024-04-08 01:30:16 -07:00
parent f33743c29d
commit 303eb29ece
1 changed files with 1 additions and 1 deletions
@@ -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: