salesforce/BLIP
1
0
Fork 0
mirror of https://github.com/salesforce/BLIP.git synced 2026-02-22 03:53:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

update demo

Browse Source
This commit is contained in:
root
2022-01-29 02:41:17 +00:00
parent 7e376065fd
commit a176f1e9cc
2 changed files with 52 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
demo.ipynb
View File

File diff suppressed because one or more lines are too long

2
models/blip.py
View File

@@ -23,7 +23,7 @@ from timm.models.hub import download_cached_file
class BLIP_Base(nn.Module):
def __init__(self,
med_config = 'configs/med_config.json',
image_size = 384,
image_size = 224,
vit = 'base',
vit_grad_ckpt = False,
vit_ckpt_layer = 0,