Junfeng5 commited on
Commit
caf313c
·
verified ·
1 Parent(s): e0db034

Update model/multimodal_encoder/openclip_encoder.py

Browse files
model/multimodal_encoder/openclip_encoder.py CHANGED
@@ -4,7 +4,7 @@ import torch.nn.functional as F
4
  import os
5
  import json
6
  import logging
7
- import deepspeed
8
  from pathlib import Path
9
  from open_clip.factory import load_state_dict, get_model_config
10
  from open_clip.model import CLIPVisionCfg, CLIPTextCfg, _build_vision_tower, convert_to_custom_text_state_dict, resize_pos_embed
 
4
  import os
5
  import json
6
  import logging
7
+ # import deepspeed
8
  from pathlib import Path
9
  from open_clip.factory import load_state_dict, get_model_config
10
  from open_clip.model import CLIPVisionCfg, CLIPTextCfg, _build_vision_tower, convert_to_custom_text_state_dict, resize_pos_embed