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 |
|