jinaai
/

jina-clip-implementation

Inference Endpoints

🇪🇺 Region: EU

Model card Files Files and versions Community

bwang0911 commited on May 28

Commit

4ed2f34

•

1 Parent(s): fe33287

Update modeling_clip.py

Files changed (1) hide show

modeling_clip.py +1 -1

modeling_clip.py CHANGED Viewed

@@ -12,7 +12,7 @@ import torch
 import torch.nn.functional as f
 import torch.utils.checkpoint
 from torch import nn
-from transformers import BatchEncoding, BatchFeature, PreTrainedModel, logging, AutoImageProcessor
 from transformers.models.clip.modeling_clip import (
     CLIPOutput,
     CLIPTextModelOutput,

 import torch.nn.functional as f
 import torch.utils.checkpoint
 from torch import nn
+from transformers import BatchEncoding, BatchFeature, PreTrainedModel, logging, AutoImageProcessor, AutoTokenizer
 from transformers.models.clip.modeling_clip import (
     CLIPOutput,
     CLIPTextModelOutput,