mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-15 04:52:31 +08:00
typo correction
This commit is contained in:
parent
0a5a986bcf
commit
cf5faf8c17
@ -24,7 +24,7 @@ IMAGE_ENCODERS = {
|
||||
"siglip_vision_model": comfy.clip_model.CLIPVisionModelProjection,
|
||||
"siglip2_vision_model": comfy.clip_model.CLIPVisionModelProjection,
|
||||
"dinov2": comfy.image_encoders.dino2.Dinov2Model,
|
||||
"briefnet": comfy.image_encoders.birefnet.BiRefNet
|
||||
"birefnet": comfy.image_encoders.birefnet.BiRefNet
|
||||
}
|
||||
|
||||
class ClipVisionModel():
|
||||
@ -132,7 +132,7 @@ 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_vitl.json")
|
||||
|
||||
elif "bb.layers.1.blocks.0.attn.relative_position_index" in sd:
|
||||
json_config = os.path.join(os.path.join(os.path.dirname(os.path.realpath(__file__)), "image_encoders"), "briefnet.json")
|
||||
json_config = os.path.join(os.path.join(os.path.dirname(os.path.realpath(__file__)), "image_encoders"), "birefnet.json")
|
||||
|
||||
# Dinov2
|
||||
elif 'encoder.layer.39.layer_scale2.lambda1' in sd:
|
||||
|
||||
@ -1,6 +1,6 @@
|
||||
{
|
||||
"model_type": "briefnet",
|
||||
"model_type": "birefnet",
|
||||
"image_std": [1.0, 1.0, 1.0],
|
||||
"image_mean": [0.0, 0.0, 0.0],
|
||||
"image_size": 1024
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user