Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fooocus GitHub Bot Commit

Browse Source 
This commit is generated by a GitHub bot of Fooocus
This commit is contained in:
lllyasviel
2023-10-19 06:32:38 -07:00
parent 6f02c797c6
commit f553d0fb18
6 changed files with 93 additions and 72 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/headless/fcbh/clip_vision.py
+4 -1
View File
@@ -92,8 +92,11 @@ def load_clipvision_from_sd(sd, prefix="", convert_keys=False):
json_config = os.path.join(os.path.dirname(os.path.realpath(__file__)), "clip_vision_config_g.json")
elif "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:
elif "vision_model.encoder.layers.22.layer_norm1.weight" in sd:
json_config = os.path.join(os.path.dirname(os.path.realpath(__file__)), "clip_vision_config_vitl.json")
else:
return None
clip = ClipVisionModel(json_config)
m, u = clip.load_sd(sd)
if len(m) > 0: