v4 vs-trt support in vsmlrt.py, RealESRGANv2 model & full binary releases
This release introduces the following features:
- vsmlrt.py: added support for vs-trt (including transparent engine compilation)
- added RealESRGANv2 models, see https://github.com/AmusementClub/vs-mlrt/releases/download/model-20211209/RealESRGANv2_v1.7z
- full binary releases for Windows, which includes full set of models (waifu2x, RealESRGANv2 and DPIR) and all required DLLs. To simplify installation, we provide two variants:
- CPU only: vsmlrt-windows-x64-cpu.v4.7z
- CPU+CUDA: vsmlrt-windows-x64-cuda.v4.7z
To install, just extract them into your VSplugins
directory (preserving the existing directory structure within the 7z archive), and movevsmlrt.py
into your VS pythonsite-packages
directory and you're done.
Component Downloads
Besides the full releases, each individual component also has its own release, so that users can upgrade only what has been changed:
- models: full fp32 model release 20211209, includes waifu2x, RealESRGANv2 and DPIR.
- scripts: vsmlrt.py wrapper script, extract to VS python
site-packages
directory - vsmlrt-cuda: shared CUDA DLLs for vs-ort and vs-trt
- VSOV-Windows-x64: vs-ov plugin (pure CPU backend)
- VSORT-Windows-x64: vs-ort plugin, includes both CPU and CUDA backend; CUDA backend requires vsmlrt-cuda package.
- VSTRT-Windows-x64: vs-trt plugin, requires vsmlrt-cuda package.
All component packages should be extracted to your VS plugins
directory, except for scripts.v4.7z
, which needs to be extracted to VS python site-packages
directory.
Known Issues
- building TRT engine for waifu2x cunet and upresnet10 will fail on RTX 2000 and RTX 3000 series GPUs, please use vsort if you are using affected GPUs.
- due to the way NVidia DLLs are named, there might be DLL conflicts if you also have other AI filters (e.g. waifu2x caffe) in your plugins directory. Due to licensing restrictions and windows technical restrictions, there is no easy way to solve this DLL conflict problem. You will have to remove those conflicting plugins. Fortunately, the only affected plugin seems to be waifu2x caffe and we have already provided full functionality coverage and better performance with the vsmlrt.py script so there is no reason to use the caffe plugin anymore.
Installation Notes
- It is recommended to update to the latest GPU driver (e.g. >= v472.50) if you intend to use the CUDA backend of vsort or vstrt for best performance and compatibility; However, GeForce GPU users with GPU driver >= v452.39 should be able to use the CUDA backend.
- There are no changes to vsmlrt-cuda.7z from v3, so no need to re-download it if you already have it from v3.