load clip encoder from path if possible

clip_encoder.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 import open_clip
 from einops import rearrange
-
+import os
 
 def exists(val):
     return val is not None
@@ -11,6 +11,12 @@ class CLIPEncoder(nn.Module):
 
     def __init__(self, model, pretrained):
         super().__init__()
+        #ViT_H_14_laion2b_s32b_b79k
+        fname = "models/" + model.replace("-", "_") + "_" + pretrained + ".pt"
+
+        if os.path.exists(fname):
+            print(fname)
+            pretrained = fname
         self.model = model
         self.pretrained = pretrained
         self.model, _, _ = open_clip.create_model_and_transforms(model, pretrained=pretrained)