Spaces:
Runtime error
Runtime error
add location
Browse files
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
|