andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix error with ClipVision loader node.

Browse Source
This commit is contained in:
comfyanonymous
2023-06-23 01:08:05 -04:00
parent 3e0686ce94
commit 9e37f4c7d5
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/clip_vision.py
View File

@@ -52,8 +52,9 @@ def convert_to_transformers(sd, prefix):
sd = transformers_convert(sd, prefix, "vision_model.", 32)
return sd
def load_clipvision_from_sd(sd, prefix):
sd = convert_to_transformers(sd, prefix)
def load_clipvision_from_sd(sd, prefix="", convert_keys=False):
if convert_keys:
sd = convert_to_transformers(sd, prefix)
if "vision_model.encoder.layers.30.layer_norm1.weight" in sd:
json_config = os.path.join(os.path.dirname(os.path.realpath(__file__)), "clip_vision_config_h.json")
else: