From 16278b5416ef74be82236fe9ae04a0c8fe9ce6f7 Mon Sep 17 00:00:00 2001
From: Lannie Hough <lannie.hough@protonmail.com>
Date: Wed, 18 Jun 2025 14:29:24 -0400
Subject: [PATCH] correctly import find_multiple from litgpt version >= 0.5.9

---
 yalis/external/config.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/yalis/external/config.py b/yalis/external/config.py
index 22c15cba..e79068fe 100644
--- a/yalis/external/config.py
+++ b/yalis/external/config.py
@@ -8,8 +8,17 @@
 import torch
 import yaml
 from typing_extensions import Self
-from litgpt.utils import find_multiple
 
+from packaging.version import Version
+from importlib.metadata import version, PackageNotFoundError
+try:
+    litgpt_version = version("litgpt")
+except:
+    raise RuntimeError("litgpt isn't installed!")
+if Version(litgpt_version) >= Version("0.5.9"):
+    from litgpt.config import find_multiple
+else:
+    from litgpt.utils import find_multiple
 
 @dataclass
 class Config: