pip install onnx onnxruntime-gpu
# Load the model model = torch.load('gpen-bfr-2048.pth', map_location=torch.device('cpu'))
Often used in tandem with colorization scripts to bring black-and-white portraits to life.
pip install onnx onnxruntime-gpu
# Load the model model = torch.load('gpen-bfr-2048.pth', map_location=torch.device('cpu')) gpen-bfr-2048.pth
Often used in tandem with colorization scripts to bring black-and-white portraits to life. pip install onnx onnxruntime-gpu # Load the model