From c490b9cb51fe24c28c0232d0e86f565b1d07ab4c Mon Sep 17 00:00:00 2001 From: Seunghoon Lee Date: Fri, 9 Aug 2024 00:45:12 +0900 Subject: [PATCH] fix first launch --- modules/rocm.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/rocm.py b/modules/rocm.py index 120d5582a..5eab4f00a 100644 --- a/modules/rocm.py +++ b/modules/rocm.py @@ -83,7 +83,11 @@ class Agent: def get_version_torch() -> Union[str, None]: - version_ = importlib.metadata.version("torch") + version_ = None + try: + version_ = importlib.metadata.version("torch") + except importlib.metadata.PackageNotFoundError: + return None if "+rocm" not in version_: # unofficial build, non-rocm torch. return None return version_.split("+rocm")[1]