diff --git a/README.md b/README.md index 05d9bdc..9d436b8 100644 --- a/README.md +++ b/README.md @@ -114,7 +114,7 @@ Clone into your ComfyUI `custom_nodes/` directory: ```bash cd ComfyUI/custom_nodes -git clone https://github.com/your-user/Comfyui-BIM-VFI.git +git clone https://github.com/your-user/ComfyUI-Tween.git ``` Dependencies (`gdown`, `cupy`, `timm`) are auto-installed on first load. The correct `cupy` variant is detected from your PyTorch CUDA version. @@ -127,7 +127,7 @@ Dependencies (`gdown`, `cupy`, `timm`) are auto-installed on first load. The cor To install manually: ```bash -cd Comfyui-BIM-VFI +cd ComfyUI-Tween python install.py ``` diff --git a/__init__.py b/__init__.py index af2d6f1..5cc94a4 100644 --- a/__init__.py +++ b/__init__.py @@ -2,7 +2,7 @@ import subprocess import sys import logging -logger = logging.getLogger("BIM-VFI") +logger = logging.getLogger("Tween") def _auto_install_deps(): @@ -11,14 +11,14 @@ def _auto_install_deps(): try: import gdown # noqa: F401 except ImportError: - logger.info("[BIM-VFI] Installing gdown...") + logger.info("[Tween] Installing gdown...") subprocess.check_call([sys.executable, "-m", "pip", "install", "gdown"]) # timm (required for EMA-VFI's MotionFormer backbone) try: import timm # noqa: F401 except ImportError: - logger.info("[BIM-VFI] Installing timm...") + logger.info("[Tween] Installing timm...") subprocess.check_call([sys.executable, "-m", "pip", "install", "timm"]) # cupy @@ -29,10 +29,10 @@ def _auto_install_deps(): import torch major = int(torch.version.cuda.split(".")[0]) cupy_pkg = f"cupy-cuda{major}x" - logger.info(f"[BIM-VFI] Installing {cupy_pkg} (CUDA {torch.version.cuda})...") + logger.info(f"[Tween] Installing {cupy_pkg} (CUDA {torch.version.cuda})...") subprocess.check_call([sys.executable, "-m", "pip", "install", cupy_pkg]) except Exception as e: - logger.warning(f"[BIM-VFI] Could not auto-install cupy: {e}") + logger.warning(f"[Tween] Could not auto-install cupy: {e}") _auto_install_deps() diff --git a/bim_vfi_arch/costvol.py b/bim_vfi_arch/costvol.py index 931c6d0..93bd5d9 100644 --- a/bim_vfi_arch/costvol.py +++ b/bim_vfi_arch/costvol.py @@ -18,7 +18,7 @@ def _ensure_cupy(): raise RuntimeError( "cupy is required for BIM-VFI. Install it with:\n" " pip install cupy-cuda12x (or cupy-cuda11x for CUDA 11)\n" - "Or run install.py from the Comfyui-BIM-VFI directory." + "Or run install.py from the ComfyUI-Tween directory." ) diff --git a/inference.py b/inference.py index 01203c4..33cf372 100644 --- a/inference.py +++ b/inference.py @@ -11,7 +11,7 @@ from .sgm_vfi_arch import feature_extractor as sgm_feature_extractor from .sgm_vfi_arch import MultiScaleFlow as SGMMultiScaleFlow from .utils.padder import InputPadder -logger = logging.getLogger("BIM-VFI") +logger = logging.getLogger("Tween") class BiMVFIModel: diff --git a/install.py b/install.py index 7b478a2..fe28047 100644 --- a/install.py +++ b/install.py @@ -8,18 +8,18 @@ def get_cupy_package(): try: import torch if not torch.cuda.is_available(): - print("[BIM-VFI] WARNING: CUDA not available. cupy requires CUDA.") + print("[Tween] WARNING: CUDA not available. cupy requires CUDA.") return None cuda_version = torch.version.cuda if cuda_version is None: - print("[BIM-VFI] WARNING: PyTorch has no CUDA version info.") + print("[Tween] WARNING: PyTorch has no CUDA version info.") return None major = int(cuda_version.split(".")[0]) cupy_pkg = f"cupy-cuda{major}x" - print(f"[BIM-VFI] Detected CUDA {cuda_version}, will use {cupy_pkg}") + print(f"[Tween] Detected CUDA {cuda_version}, will use {cupy_pkg}") return cupy_pkg except Exception as e: - print(f"[BIM-VFI] WARNING: Could not detect CUDA version: {e}") + print(f"[Tween] WARNING: Could not detect CUDA version: {e}") return None diff --git a/nodes.py b/nodes.py index 2358a17..fe59495 100644 --- a/nodes.py +++ b/nodes.py @@ -13,7 +13,7 @@ from .bim_vfi_arch import clear_backwarp_cache from .ema_vfi_arch import clear_warp_cache as clear_ema_warp_cache from .sgm_vfi_arch import clear_warp_cache as clear_sgm_warp_cache -logger = logging.getLogger("BIM-VFI") +logger = logging.getLogger("Tween") # Google Drive file ID for the pretrained BIM-VFI model GDRIVE_FILE_ID = "18Wre7XyRtu_wtFRzcsit6oNfHiFRt9vC"