Cleanup.

Author: comfyanonymous

comfy/model_management.py

@@ -45,6 +45,7 @@ class CPUState(Enum):
 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()

server.py

@@ -30,7 +30,6 @@
 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 @@ async def system_stats(request):
                     "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
                 },