update demo

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

View File

@@ -9,7 +9,7 @@ import numpy as np
class BLIP_VQA(nn.Module):
def __init__(self,
med_config = './configs/med_config.json',
med_config = 'configs/med_config.json',
image_size = 480,
vit = 'base',
vit_grad_ckpt = False,