Shaoan commited on
Commit
fd9e0dc
·
verified ·
1 Parent(s): 322babc

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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 = './clip-vit-large-patch14'
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)