adding logging

2026-04-23 12:08:58 -04:00
parent e4150f8e93
commit 2464f44f74
3 changed files with 11 additions and 4 deletions
@@ -18,6 +18,10 @@ import torch
 import comfy.model_management as mm

 from pathlib import Path
+
+# Use polygraphy TensorRTX backend
+os.environ["POLYGRAPHY_USE_TENSORRT_RTX"] = "1"
+
 from polygraphy import cuda
 from .vfi_utilities import preprocess_frames, postprocess_frames, generate_frames_rife
 from .trt_utilities import Engine