search

yuvraj108c / ComfyUI-Upscaler-Tensorrt

3-4x faster ComfyUI Image Upscaling using Tensorrt
Other
64 stars 4 forks source link

Converting more NMKD ESRGAN Models to ONNX #28

Open thomasza92 opened 2 days ago

thomasza92 commented 2 days ago

Is it possible to convert 4x_NMKD-Yandere4_120000_G.pth and 4x_NMKD-YandereNeoXL_200k.pth models to onnx? If so what is procedure? Or can you add to repository? Thank you for all your hard work!

yuvraj108c commented 2 days ago

  1. Download the model into comfyui models/upscale_models

  2. Modify the following lines in export_onnx.py

https://github.com/yuvraj108c/ComfyUI-Upscaler-Tensorrt/blob/bdad3d789c1cc9e80b6f42e27870d2f64d050f82/export_onnx.py#L100C6-L100C48

https://github.com/yuvraj108c/ComfyUI-Upscaler-Tensorrt/blob/bdad3d789c1cc9e80b6f42e27870d2f64d050f82/export_onnx.py#L115C1-L115C61

  1. Move export_onnx.py file to comfyui root directory, run the file, then build the tensorrt engine from the generated onnx file