Spaces:
Running
on
Zero
Running
on
Zero
Upload folder using huggingface_hub
Browse files- text_encoder.py +1 -1
text_encoder.py
CHANGED
|
@@ -420,7 +420,7 @@ class LoraT5Embedder(torch.nn.Module):
|
|
| 420 |
|
| 421 |
print(f"Gradient checkpointing enabled: {self.use_gradient_checkpointing}")
|
| 422 |
|
| 423 |
-
image_encoder_path = '
|
| 424 |
self.image_encoder = CLIPVisionModelWithProjection.from_pretrained(
|
| 425 |
image_encoder_path
|
| 426 |
).to(device=device).to(torch.bfloat16)
|
|
|
|
| 420 |
|
| 421 |
print(f"Gradient checkpointing enabled: {self.use_gradient_checkpointing}")
|
| 422 |
|
| 423 |
+
image_encoder_path = 'openai/clip-vit-large-patch14'
|
| 424 |
self.image_encoder = CLIPVisionModelWithProjection.from_pretrained(
|
| 425 |
image_encoder_path
|
| 426 |
).to(device=device).to(torch.bfloat16)
|