xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-17 13:36:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set clip vision model in eval mode so it works without inference mode.

Browse Source
This commit is contained in:
comfyanonymous 2023-12-15 18:53:08 -05:00
parent 6a3d15681b
commit 5b1e4dff6c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/clip_vision.py
View File

@ -39,6 +39,7 @@ class ClipVisionModel():
self.dtype = torch.float16
self.model = comfy.clip_model.CLIPVisionModelProjection(config, self.dtype, offload_device, comfy.ops.disable_weight_init)
self.model.eval()
self.patcher = comfy.model_patcher.ModelPatcher(self.model, load_device=self.load_device, offload_device=offload_device)
def load_sd(self, sd):