Commit Graph
25 Commits
Author SHA1 Message Date
huchukatoandDevin 970f51649b Add RIFE v4.25 support, clean dropdown, pin polygraphy
- Add RIFE v4.25 architecture support in rife_arch.py (Head with
  16/4 channels, 5 IFBlocks with c=192/128/96/64/32, scale_list
  [16,8,4,2,1], no ensemble)
- Add rife425.pth to export_onnx.py with HF fallback URL
- Update load_rife_config.json: dropdown now only shows rife425 and
  rife49 (removed rife47/48/417/426/sudo_rife4 that never worked)
- Set rife425 as default
- Make TORCH_DEVICE fallback to CPU for ONNX export on non-CUDA
- Pin polygraphy==0.49.26 for TensorRT 10.15.1 compatibility

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

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-28 02:50:00 +02:00
huchukatoandDevin 2132614242 Add huchukato/favs as primary ONNX download source
Use the user's own HuggingFace repo as the first source for ONNX
models, falling back to yuvraj108c and marduk191 mirrors.

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

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-14 20:14:36 +02:00
huchukatoandDevin def1bea883 Add ONNX download validation and multi-source fallback
The download_file() function now validates:
- HTTP status code (fails on non-200)
- File size (removes files <1KB as likely error pages)
- ONNX header (detects HTML pages masquerading as .onnx files)

The ONNX download in AutoLoadRifeTensorrtModel now tries multiple
HuggingFace sources before giving up, with clear error message
pointing users to manual export_onnx.py as fallback.

This prevents the cryptic "factory function returned nullptr" error
when TensorRT attempts to parse a corrupted/empty/HTML file as ONNX.

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

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-14 20:03:29 +02:00
huchukatoandDevin 120f2ca1b1 Fix auto-install reliability and bump version to 1.2.0
- Verify tensorrt is actually importable before trusting the install marker;
  retry install if marker exists but import fails (likely cause of user reports
  where the node loaded but TensorRT did not work)
- Add --no-input to pip install calls to avoid hanging on interactive prompts
- Show pip stdout during auto-install so users can see progress
- Improve failed-install cooldown message with retry instructions
- Bump version to 1.2.0

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

Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
2026-08-12 11:52:53 +02:00
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