tennant commited on
Commit
71e1a05
1 Parent(s): b14c338

add location

Browse files
Files changed (1) hide show
  1. model.py +1 -2
model.py CHANGED
@@ -430,8 +430,7 @@ class MaskedAutoencoderViT(nn.Module):
430
  # captioner specifics
431
  # unimodal layer is for text tokens.
432
  # multimodal layer is for text to query from image.
433
- self.tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased",
434
- cache_dir='/disk/scratch_fast/bingchen/.cache/torch/hub/checkpoints/bert-base-uncased', local_files_only=True)
435
 
436
  # token embeddings
437
  # NOTE: +1 for mask token used by MLM objective
430
  # captioner specifics
431
  # unimodal layer is for text tokens.
432
  # multimodal layer is for text to query from image.
433
+ self.tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased", )
 
434
 
435
  # token embeddings
436
  # NOTE: +1 for mask token used by MLM objective