kellempxt/ComfyUI-Facerestore-Tensorrt

By kellempxt · ↓ 0 · ♥ 0 · Updated 2025-01-16

What's in this repo

This page lists the files in kellempxt/ComfyUI-Facerestore-Tensorrt, including codeformer.onnx, gfqgan.onnx.

2 ONNX models (.onnx files)
6 files, including 2 .engine, 2 .onnx, 1 .md
Last updated 2025-01-16

ONNX files

File	Size	Download
codeformer/codeformer.onnx	—	Download
gfqgan/gfqgan.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

ComfyUI-Facerestore-Tensorrt

https://github.com/yuvraj108c/ComfyUI-Facerestore-Tensorrt/tree/master

The instruction is to simply download the following onnx models:

gfqgan.onnx
codeformer.onnx

And build tensorrt engines for these models by running:

python export_trt.py

These files should work if you are using NVIDIA AGX ORIN 64GB Dev Kit. Otherwise you will need to build your own engine file as it is dependant on your hardware.

Good Luck~!

Model info

onnx region:us

↓ 0 ♥ 0

Created: 2025-01-12

Updated: 2025-01-16

View on Hugging Face

Files (6)

.gitattributes

README.md

codeformer/codeformer.engine

codeformer/codeformer.onnx ONNX

gfqgan/gfqgan.engine

gfqgan/gfqgan.onnx ONNX

Related models

More models

No task type declared.