warmshao / FasterLivePortrait

Bring portraits to life in Real Time!onnx/tensorrt support!实时肖像驱动!
455 stars 42 forks source link

ERROR:EngineBuilder:Failed to load ONNX file: /home/FasterLivePortrait/checkpoints/liveportrait_onnx/warping_spade-fix.onnx #35

Closed linhcentrio closed 1 month ago

linhcentrio commented 1 month ago

(/home/Efficient-Live-Portrait/venv) [root@aiclip-dfl FasterLivePortrait]# python scripts/onnx2trt.py -o ./checkpoints/liveportrait_onnx/warping_spade-fix.onnx [07/25/2024-08:48:52] [TRT] [I] [MemUsageChange] Init CUDA: CPU +12, GPU +0, now: CPU 27, GPU 160 (MiB) [07/25/2024-08:49:05] [TRT] [I] [MemUsageChange] Init builder kernel library: CPU +1226, GPU +268, now: CPU 1330, GPU 428 (MiB) /home/FasterLivePortrait/scripts/onnx2trt.py:57: DeprecationWarning: Use set_memory_pool_limit instead. self.config.max_workspace_size = 12 * (2 ** 30) # 12 GB [07/25/2024-08:49:06] [TRT] [W] onnx2trt_utils.cpp:374: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32. [07/25/2024-08:49:06] [TRT] [I] No importer registered for op: GridSample3D. Attempting to import as plugin. [07/25/2024-08:49:06] [TRT] [I] Searching for plugin: GridSample3D, plugin_version: 1, plugin_namespace: [07/25/2024-08:49:06] [TRT] [E] 3: getPluginCreator could not find plugin: GridSample3D version: 1 ERROR:EngineBuilder:Failed to load ONNX file: /home/FasterLivePortrait/checkpoints/liveportrait_onnx/warping_spade-fix.onnx ERROR:EngineBuilder:In node 11 (importFallbackPluginImporter): UNSUPPORTED_NODE: Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?"

warmshao commented 1 month ago

Efficient-Live-Portrait/venv ? not my project