This commit is contained in:
lllyasviel
2024-01-25 12:19:50 -08:00
parent 668cef32e4
commit 1e7becbeb6
2 changed files with 3 additions and 3 deletions

View File

@@ -12,7 +12,7 @@ import ldm_patched.modules.model_management as model_management
def get_learned_conditioning(self: sgm.models.diffusion.DiffusionEngine, batch: prompt_parser.SdConditioning | list[str]):
model_management.load_model_gpu(self.clip_patcher)
model_management.load_model_gpu(self.forge_objects.clip.patcher)
for embedder in self.conditioner.embedders:
embedder.ucg_rate = 0.0