From 6eea03538d9822aeb11c5676535be8fe37a36991 Mon Sep 17 00:00:00 2001 From: huchenlei Date: Fri, 28 Feb 2025 13:43:43 -0500 Subject: [PATCH] Use frontend pypi package --- app/frontend_management.py | 5 ++++- requirements.txt | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/app/frontend_management.py b/app/frontend_management.py index 6f20e439c..c782eef58 100644 --- a/app/frontend_management.py +++ b/app/frontend_management.py @@ -5,13 +5,16 @@ import os import re import tempfile import zipfile +import importlib from dataclasses import dataclass from functools import cached_property from pathlib import Path from typing import TypedDict, Optional import requests +from comfyui_frontend import static from typing_extensions import NotRequired + from comfy.cli_args import DEFAULT_VERSION_STRING @@ -109,7 +112,7 @@ def download_release_asset_zip(release: Release, destination_path: str) -> None: class FrontendManager: - DEFAULT_FRONTEND_PATH = str(Path(__file__).parents[1] / "web") + DEFAULT_FRONTEND_PATH = importlib.resources.path(static, ".") CUSTOM_FRONTENDS_ROOT = str(Path(__file__).parents[1] / "web_custom_versions") @classmethod diff --git a/requirements.txt b/requirements.txt index afbcb7cba..db3f17f81 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ +comfyui_frontend_package==1.10.14 torch torchsde torchvision