Spaces:

yfyangd
/

PictureBookUnderstanding

Runtime error

yfyangd commited on Sep 2, 2022

Commit

763c0b2

1 Parent(s): f976bf4

Update BLIP/models/blip.py

Files changed (1) hide show

BLIP/models/blip.py CHANGED Viewed

@@ -22,7 +22,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 = 224,
                  vit = 'base',
                  vit_grad_ckpt = False,
@@ -77,7 +77,7 @@ class BLIP_Base(nn.Module):
 class BLIP_Decoder(nn.Module):
     def __init__(self,
-                 med_config = 'configs/med_config.json',
                  image_size = 384,
                  vit = 'base',
                  vit_grad_ckpt = False,

 class BLIP_Base(nn.Module):
     def __init__(self,
+                 med_config = 'BLIP/configs/med_config.json',
                  image_size = 224,
                  vit = 'base',
                  vit_grad_ckpt = False,
 class BLIP_Decoder(nn.Module):
     def __init__(self,
+                 med_config = 'BLIP/configs/med_config.json',
                  image_size = 384,
                  vit = 'base',
                  vit_grad_ckpt = False,