Commit Graph
21 Commits
Author SHA1 Message Date
huchukatoandDevin 51d88c0c0d Add tiling, batch inference, interpolation states and multiplier schedule
Ported and adapted from Fannovel16/ComfyUI-Frame-Interpolation:

- gc.collect() after soft_empty_cache() to reduce RAM fragmentation
- Explicit float32 cast in postprocess_frames for fp16 engine safety
- InterpolationStateList class + MakeInterpolationStateList node to
  skip/keep specific frame pairs during interpolation
- Per-pair multiplier schedule (comma-separated list) overriding uniform
  multiplier, with automatic padding to match the number of pairs
- Automatic tiling with cosine feathering when frame dimensions exceed
  the TensorRT engine optimization profile (fixes 1440x2560 with max=1920)
- Padding with replicate mode when frame dimensions are below profile min
- get_input_profile_bounds() on Engine to retrieve min/opt/max shapes
- Batch inference: process multiple (pair, timestep) tasks per GPU call
  with dynamic batch axis in ONNX export and engine profile
- Backward compatible: all new features are opt-in via optional inputs

Generated with [Devin](https://devin.ai)

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-05 10:32:06 +02:00
huchukatoandDevin 1427bdef9f Reduce install time and fix registry metadata
- Auto-install now only installs the matching TensorRT wheels instead of the
  multi-GB cuda-toolkit package from PyPI.
- Use a single pip install -r call per requirements file for faster dependency
  resolution and better cache reuse.
- Add install-success/failure markers to avoid repeating the install on every
  ComfyUI startup.
- Fix pyproject.toml Repository and PublisherId to point to
  huchukato/ComfyUI-RIFE-TensorRT-Auto.

Generated with [Devin](https://devin.ai)

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-04 05:43:41 +02:00
huchukato b3e416cd27 Improve TensorRT detection with retry mechanism to handle concurrent installations 2026-03-05 07:21:52 +01:00
huchukato 876f60880e Improve CUDA version detection - use PyTorch as primary method for cloud platforms 2026-02-28 08:43:21 +01:00
huchukato cec6660e41 Remove use_cuda_graph parameter and enable automatically based on CUDA version only 2026-02-28 07:49:21 +01:00
huchukato 0f89c6a6e2 Simplify auto-installation logic - remove complex dummy module system that was causing issues 2026-02-27 10:13:50 +01:00
huchukato e4e24a9ef5 Update node categories to '⚡️ TensorRT/RIFE' with lightning icon and proper subfolder organization 2026-02-27 08:39:00 +01:00
huchukato 989e9cca61 Refine cloud detection - only disable CUDA graph for CUDA 12 in cloud environments, keep enabled for CUDA 13 2026-02-27 02:37:49 +01:00
huchukato de53234631 Auto-disable CUDA graph in cloud environments - detect RunPod, Colab, Docker and disable by default for stability 2026-02-27 02:37:13 +01:00
huchukato 3cdf12ab66 Implement fully automatic CUDA detection and TensorRT installation on node load 2026-02-21 09:14:05 +01:00
silveroxides 46d7a50445 update default resolutions and bump version to 1.1.1 2026-01-10 00:14:36 +01:00
silveroxides 5cff7f7dc7 feat: CUDA 13 default, auto-detection, and resolution profiles
Requirements updates:

- Default to CUDA 13 with TensorRT 10.14.1.48 (requirements.txt)

- Add cuda-toolkit version constraints (>=13.0.0,<13.1.0)

- Rename requirements_cu13.txt -> requirements_cu12.txt for legacy support

- Add cuda-toolkit constraint for CUDA 12 (>=12.8.0,<13.0.0)

CUDA toolkit detection:

- Add _setup_cuda_dll_path() to auto-detect CUDA installation

- Check CUDA_PATH, CUDA_HOME, and default Windows install paths

- Handle CUDA 13+ bin/x64 subdirectory for cudart64.dll

- Display download link if CUDA not found

Engine building updates:

- Add resolution profiles (small: 480-896px, medium: 720-1280px)

- Add 'custom' profile with CustomResolutionConfig node

- Include profile name in engine filename for separate caching

- Reduce default VRAM usage from ~15GB to ~2-3GB
2026-01-09 22:44:36 +01:00
Silver 45f84019b7 Update __init__.py 2025-11-29 20:31:14 +01:00
reaperhammer 25d6c9f1d2 Make the node auto download/build models and update readme.md accordingly 2025-09-25 00:08:26 +12:00
yuvraj108c 08fa66cd1e log new frames count 2024-10-04 13:36:16 +04:00
yuvraj108c 987334d48d fix cuda graph 2024-10-04 13:32:57 +04:00
yuvraj108c 7491b548d1 cuda grapsh 2024-10-04 13:30:05 +04:00
yuvraj108c 116ce22440 keep_model_loaded 2024-10-03 23:11:13 +04:00
yuvraj108c 37c7b9df6a readme 2024-10-03 08:09:08 +00:00
yuvraj108c c6aed0e8a5 cleanup 2024-10-03 07:17:18 +00:00
yuvraj108c 4952023d35 v1 2024-10-03 07:14:29 +00:00