ComfyUI-Upscaler-Tensorrt icon indicating copy to clipboard operation
ComfyUI-Upscaler-Tensorrt copied to clipboard

readme is unclear (update it with more detail)

Open fractalfantasy opened this issue 9 months ago • 5 comments

Took me a few days to understand how to build the engine, so im putting this here to help other people:

  1. git clone the repo to your custom_nodes directory

  2. navigate command prompt to ComfyUI-Upscaler-Tensorrt directory and install it: pip install -r requirements.txt

  3. download onnx files to this directory: ..\ComfyUI\app\models\tensorrt\upscaler\onnx

  4. run engine builder script from ComfyUI-Upscaler-Tensorrt directory to build the engine. python export_trt.py Enter the path to save the TensorRT engine (e.g ./realesrgan.engine): ..\ComfyUI\app\models\tensorrt\upscaler\4x_foolhardy_Remacri.engine Enter the path to the ONNX model (e.g ./realesrgan.onnx): ..\ComfyUI\app\models\tensorrt\upscaler\onnx\4x_foolhardy_Remacri.onnx

  • do not create a .engine directory/file before you run the script, it will automatically create this file for you. just give it the filepath for the future .engine file , see my example above
  • there should not be quotes '' around the paths

once the engine(s) are built!!

  1. open the node in ComfyUi and place it at the end of your IMAGE chain

right click >> add node >> tensorrt >> Upscale Tensorrt

image
  1. pick your .engine file and run it, it should upscale the image :)

fractalfantasy avatar May 27 '24 23:05 fractalfantasy