huchukatoandDevin
2e1c533e34
Update rife_arch.py with support for new architectures: - v4.17: Head_417 encoder, IFBlock with 16 extra channels - v4.26: 5 IFBlocks (block4 extra), lastconv 4*13 channels, feat_out return, no ensemble, 5 scale levels - Head/Head_417 classes for v4.17/v4.26 encoders - warp() fix: MPS padding_mode, dtype mismatch, get_torch_device() Update export_onnx.py: - Add rife417.pth and rife426.pth to CKPT_NAME_VER_DICT - Add CKPT_FALLBACK_URLS with HuggingFace mirrors for all models - Add export calls for rife49, rife417, rife426, sudo_rife4 Update load_rife_config.json with 3 new ONNX model options. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
29 lines
1.2 KiB
JSON
29 lines
1.2 KiB
JSON
{
|
|
"model": {
|
|
"options": [
|
|
"rife47_ensemble_True_scale_1_sim",
|
|
"rife48_ensemble_True_scale_1_sim",
|
|
"rife49_ensemble_True_scale_1_sim",
|
|
"rife417_ensemble_True_scale_1_sim",
|
|
"rife426_ensemble_False_scale_1_sim",
|
|
"sudo_rife4_269.662_testV1_scale1_ensemble_True_scale_1_sim"
|
|
],
|
|
"default": "rife49_ensemble_True_scale_1_sim",
|
|
"tooltip": "RIFE models for video frame interpolation. rife49 (v4.7) is the stable default. rife417 (v4.17) and rife426 (v4.26) are newer with improved quality. sudo_rife4 is a community-trained v4.0 variant."
|
|
},
|
|
"precision": {
|
|
"options": ["fp16", "fp32"],
|
|
"default": "fp16",
|
|
"tooltip": "Precision to build the tensorrt engines. Loaded from config."
|
|
},
|
|
"resolution_profile": {
|
|
"options": ["small", "medium", "large"],
|
|
"default": "small",
|
|
"tooltip": "Resolution range for TensorRT engine. Small: 384-1080px, Medium: 672-1312px, Large: 720-1920px. Smaller ranges use less VRAM."
|
|
},
|
|
"resolution_profiles": {
|
|
"small": { "min": 384, "opt": 720, "max": 1080 },
|
|
"medium": { "min": 672, "opt": 1080, "max": 1312 },
|
|
"large": { "min": 720, "opt": 1440, "max": 1920 }
|
|
}
|
|
} |