From 935ae153e154813ace36db4c4656a5e96f403eba Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 30 Aug 2024 12:48:42 -0400 Subject: [PATCH] Cleanup. --- comfy/model_management.py | 1 + server.py | 5 ++--- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/model_management.py b/comfy/model_management.py index 4147989e..59dce4da 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -45,6 +45,7 @@ cpu_state = CPUState.GPU total_vram = 0 xpu_available = False +torch_version = "" try: torch_version = torch.version.__version__ xpu_available = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available() diff --git a/server.py b/server.py index 26bfcdd9..3611880d 100644 --- a/server.py +++ b/server.py @@ -30,7 +30,6 @@ from app.user_manager import UserManager from model_filemanager import download_model, DownloadModelStatus from typing import Optional from api_server.routes.internal.internal_routes import InternalRoutes -import torch class BinaryEventTypes: PREVIEW_IMAGE = 1 @@ -52,7 +51,7 @@ def get_comfyui_version(): except Exception: try: import subprocess - comfyui_version = subprocess.check_output(["git", "describe", "--tags"], cwd=repo_path) + comfyui_version = subprocess.check_output(["git", "describe", "--tags"], cwd=repo_path).decode('utf-8') except Exception as e: logging.warning(f"Failed to get ComfyUI version: {e}") return comfyui_version.strip() @@ -427,7 +426,7 @@ class PromptServer(): "os": os.name, "comfyui_version": get_comfyui_version(), "python_version": sys.version, - "pytorch_version": torch.version.__version__, + "pytorch_version": comfy.model_management.torch_version, "embedded_python": os.path.split(os.path.split(sys.executable)[0])[1] == "python_embeded", "argv": sys.argv },