Make sure to fully load model in GPU memory.

2024-09-19 01:29:23 -04:00
parent ffd330d83a
commit b3eb08f1b2
1 changed files with 1 additions and 1 deletions
@@ -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)