salesforce/BLIP
1
0
Fork 0
mirror of https://github.com/salesforce/BLIP.git synced 2026-04-29 20:01:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

update demo

Browse Source
This commit is contained in:
root
2022-01-28 12:59:36 +00:00
parent 48e4eba7a9
commit f68d24cf16
6 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/blip_pretrain.py
View File

@@ -18,7 +18,7 @@ from models.blip import create_vit, init_tokenizer, load_checkpoint
class BLIP_Pretrain(nn.Module):
def __init__(self,
med_config = './configs/bert_config.json',
med_config = 'configs/bert_config.json',
image_size = 224,
vit = 'base',
vit_grad_ckpt = False,