yyk19 commited on
Commit
ee736c7
1 Parent(s): 84289e0
Files changed (1) hide show
  1. ldm/modules/encoders/modules.py +3 -1
ldm/modules/encoders/modules.py CHANGED
@@ -205,8 +205,10 @@ class FrozenOpenCLIPEmbedder(AbstractEncoder):
205
  freeze=True, layer="last"):
206
  super().__init__()
207
  assert layer in self.LAYERS
 
208
  model, _, _ = open_clip.create_model_and_transforms(arch, device=torch.device('cpu'), pretrained=version)
209
- aa = model.encode_image(torch.zeros((1, 3,224,224)))
 
210
  del model.visual
211
  self.model = model
212
 
 
205
  freeze=True, layer="last"):
206
  super().__init__()
207
  assert layer in self.LAYERS
208
+ print("Start initializing the CLIP text encoder")
209
  model, _, _ = open_clip.create_model_and_transforms(arch, device=torch.device('cpu'), pretrained=version)
210
+ print("Initialization ends")
211
+ # aa = model.encode_image(torch.zeros((1, 3,224,224)))
212
  del model.visual
213
  self.model = model
214