Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename project to ComfyUI-Tween

Browse Source 
Update logger names, install prefixes, README clone instructions, and
error messages to reflect the new repo name. Model-specific node names
and categories (BIM-VFI, EMA-VFI, SGM-VFI) are unchanged.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-12 23:08:54 +01:00
parent 42ebdd8b96
commit e37cc3dd2e
6 changed files with 14 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -114,7 +114,7 @@ Clone into your ComfyUI `custom_nodes/` directory:
```bash ```bash
cd ComfyUI/custom_nodes 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. 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: To install manually:
```bash ```bash
cd Comfyui-BIM-VFI cd ComfyUI-Tween
python install.py python install.py
``` ```

10
__init__.py
View File

@@ -2,7 +2,7 @@ import subprocess
import sys import sys
import logging import logging
logger = logging.getLogger("BIM-VFI") logger = logging.getLogger("Tween")
def _auto_install_deps(): def _auto_install_deps():
@@ -11,14 +11,14 @@ def _auto_install_deps():
try: try:
import gdown # noqa: F401 import gdown # noqa: F401
except ImportError: except ImportError:
logger.info("[BIM-VFI] Installing gdown...") logger.info("[Tween] Installing gdown...")
subprocess.check_call([sys.executable, "-m", "pip", "install", "gdown"]) subprocess.check_call([sys.executable, "-m", "pip", "install", "gdown"])
# timm (required for EMA-VFI's MotionFormer backbone) # timm (required for EMA-VFI's MotionFormer backbone)
try: try:
import timm # noqa: F401 import timm # noqa: F401
except ImportError: except ImportError:
logger.info("[BIM-VFI] Installing timm...") logger.info("[Tween] Installing timm...")
subprocess.check_call([sys.executable, "-m", "pip", "install", "timm"]) subprocess.check_call([sys.executable, "-m", "pip", "install", "timm"])
# cupy # cupy
@@ -29,10 +29,10 @@ def _auto_install_deps():
import torch import torch
major = int(torch.version.cuda.split(".")[0]) major = int(torch.version.cuda.split(".")[0])
cupy_pkg = f"cupy-cuda{major}x" 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]) subprocess.check_call([sys.executable, "-m", "pip", "install", cupy_pkg])
except Exception as e: 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() _auto_install_deps()

2
bim_vfi_arch/costvol.py
View File

@@ -18,7 +18,7 @@ def _ensure_cupy():
raise RuntimeError( raise RuntimeError(
"cupy is required for BIM-VFI. Install it with:\n" "cupy is required for BIM-VFI. Install it with:\n"
" pip install cupy-cuda12x (or cupy-cuda11x for CUDA 11)\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."
) )

2
inference.py
View File

@@ -11,7 +11,7 @@ from .sgm_vfi_arch import feature_extractor as sgm_feature_extractor
from .sgm_vfi_arch import MultiScaleFlow as SGMMultiScaleFlow from .sgm_vfi_arch import MultiScaleFlow as SGMMultiScaleFlow
from .utils.padder import InputPadder from .utils.padder import InputPadder
logger = logging.getLogger("BIM-VFI") logger = logging.getLogger("Tween")
class BiMVFIModel: class BiMVFIModel:

8
install.py
View File

@@ -8,18 +8,18 @@ def get_cupy_package():
try: try:
import torch import torch
if not torch.cuda.is_available(): 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 return None
cuda_version = torch.version.cuda cuda_version = torch.version.cuda
if cuda_version is None: 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 return None
major = int(cuda_version.split(".")[0]) major = int(cuda_version.split(".")[0])
cupy_pkg = f"cupy-cuda{major}x" 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 return cupy_pkg
except Exception as e: 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 return None

2
nodes.py
View File

@@ -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 .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 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 # Google Drive file ID for the pretrained BIM-VFI model
GDRIVE_FILE_ID = "18Wre7XyRtu_wtFRzcsit6oNfHiFRt9vC" GDRIVE_FILE_ID = "18Wre7XyRtu_wtFRzcsit6oNfHiFRt9vC"