Spaces:
Runtime error
Runtime error
for debug
Browse files
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 |
-
|
|
|
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 |
|