Patsorn commited on
Commit
e22ba9d
·
unverified ·
1 Parent(s): 751e6db

remove unused imports

Browse files
Files changed (1) hide show
  1. code/clip/model.py +1 -5
code/clip/model.py CHANGED
@@ -237,10 +237,6 @@ class VisualTransformer(nn.Module):
237
 
238
  return x
239
 
240
-
241
- from x_transformers.autoregressive_wrapper import AutoregressiveWrapper
242
- from x_transformers import ViTransformerWrapper, TransformerWrapper, Encoder, Decoder
243
-
244
  class CLIP(nn.Module):
245
  def __init__(self,
246
  embed_dim: int,
@@ -503,4 +499,4 @@ def build_model(state_dict: dict, weight_sharing: bool, feature_fusion: str, num
503
  convert_weights(model)
504
  #TODO: only do strict=false when loading from state with 'visual2' branch
505
  model.load_state_dict(state_dict, strict=False)
506
- return model.eval()
 
237
 
238
  return x
239
 
 
 
 
 
240
  class CLIP(nn.Module):
241
  def __init__(self,
242
  embed_dim: int,
 
499
  convert_weights(model)
500
  #TODO: only do strict=false when loading from state with 'visual2' branch
501
  model.load_state_dict(state_dict, strict=False)
502
+ return model.eval()