diff --git a/launch.py b/launch.py index 67de65a3..c32c7228 100644 --- a/launch.py +++ b/launch.py @@ -1,6 +1,8 @@ import os import sys +from modules.launch_util import commit_hash, fooocus_tag + def prepare_environment(): torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu118") @@ -12,7 +14,12 @@ def prepare_environment(): comfy_repo = os.environ.get('COMFY_REPO', "https://github.com/comfyanonymous/ComfyUI.git") comfy_commit_hash = os.environ.get('COMFY_COMMIT_HASH', "5ac96897e9782805cd5e8fe85bd98ad03eae2b6f") + commit = commit_hash() + tag = fooocus_tag + print(f"Python {sys.version}") + print(f"Version: {tag}") + print(f"Commit hash: {commit}") prepare_environment() diff --git a/modules/launch_util.py b/modules/launch_util.py new file mode 100644 index 00000000..2806ba0c --- /dev/null +++ b/modules/launch_util.py @@ -0,0 +1,17 @@ +import os + +import subprocess +from functools import lru_cache + + +git = os.environ.get('GIT', "git") +fooocus_tag = '1.0.0' + + +@lru_cache() +def commit_hash(): + try: + return subprocess.check_output([git, "rev-parse", "HEAD"], shell=False, encoding='utf8').strip() + except Exception: + return "" +