qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cleanup.

This commit is contained in:
comfyanonymous 2024-08-30 12:48:42 -04:00
parent e91662e784
commit 935ae153e1
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_management.py
View File

@ -45,6 +45,7 @@ cpu_state = CPUState.GPU
total_vram = 0 total_vram = 0
xpu_available = False xpu_available = False
torch_version = ""
try: try:
torch_version = torch.version.__version__ 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() xpu_available = (int(torch_version[0]) < 2 or (int(torch_version[0]) == 2 and int(torch_version[2]) <= 4)) and torch.xpu.is_available()

5
server.py
View File

@ -30,7 +30,6 @@ from app.user_manager import UserManager
from model_filemanager import download_model, DownloadModelStatus from model_filemanager import download_model, DownloadModelStatus
from typing import Optional from typing import Optional
from api_server.routes.internal.internal_routes import InternalRoutes from api_server.routes.internal.internal_routes import InternalRoutes
import torch
class BinaryEventTypes: class BinaryEventTypes:
PREVIEW_IMAGE = 1 PREVIEW_IMAGE = 1
@ -52,7 +51,7 @@ def get_comfyui_version():
except Exception: except Exception:
try: try:
import subprocess 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: except Exception as e:
logging.warning(f"Failed to get ComfyUI version: {e}") logging.warning(f"Failed to get ComfyUI version: {e}")
return comfyui_version.strip() return comfyui_version.strip()
@ -427,7 +426,7 @@ class PromptServer():
"os": os.name, "os": os.name,
"comfyui_version": get_comfyui_version(), "comfyui_version": get_comfyui_version(),
"python_version": sys.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", "embedded_python": os.path.split(os.path.split(sys.executable)[0])[1] == "python_embeded",
"argv": sys.argv "argv": sys.argv
}, },