remove a bug for pbc light

ykdai · Dec 4, 2024 · ea077e9 · ea077e9
1 parent 476b5c5
commit ea077e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/basicsr/archs/basicpbc_light_arch.py b/basicsr/archs/basicpbc_light_arch.py
@@ -169,7 +169,7 @@ def forward(self, line, line_ref, color_ref):
 class CLIPEncoder(nn.Module):
     def __init__(self, freeze=True):
         super().__init__()
-        clip_encoder, _, self.preprocess = open_clip.create_model_and_transforms("convnext_large_d_320")
+        clip_encoder, _, self.preprocess = open_clip.create_model_and_transforms("convnext_large_d_320", pretrained="laion2b_s29b_b131k_ft_soup")
         # We just assume preprocess is the same as our proprocess
         visual_model = clip_encoder.visual
         visual_model = list(visual_model.children())[0].children()