From b3eb08f1b2bf5c8da6629784f40bc9424b040977 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 19 Sep 2024 01:29:23 -0400 Subject: [PATCH] Make sure to fully load model in GPU memory. --- tensorrt_convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_convert.py b/tensorrt_convert.py index 5774aa3..6ca4bff 100644 --- a/tensorrt_convert.py +++ b/tensorrt_convert.py @@ -155,7 +155,7 @@ class TRT_MODEL_CONVERSION_BASE: ) comfy.model_management.unload_all_models() - comfy.model_management.load_models_gpu([model], force_patch_weights=True) + comfy.model_management.load_models_gpu([model], force_patch_weights=True, force_full_load=True) unet = model.model.diffusion_model context_dim = model.model.model_config.unet_config.get("context_dim", None)