diff --git a/comfy/clip_vision.py b/comfy/clip_vision.py index 16bd89e38..9c08b3628 100644 --- a/comfy/clip_vision.py +++ b/comfy/clip_vision.py @@ -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: diff --git a/comfy/image_encoders/briefnet.json b/comfy/image_encoders/birefnet.json similarity index 74% rename from comfy/image_encoders/briefnet.json rename to comfy/image_encoders/birefnet.json index 13b850a37..c877d549a 100644 --- a/comfy/image_encoders/briefnet.json +++ b/comfy/image_encoders/birefnet.json @@ -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 -} \ No newline at end of file +}