florentgbelidji HF staff philschmid HF staff commited on
Commit
6bacf01
1 Parent(s): 8f294b5

Update pipeline.py (#1)

Browse files

- Update pipeline.py (8085a1b108c107456579c23e3ea0fc5ecd33103c)


Co-authored-by: Philipp Schmid <philschmid@users.noreply.huggingface.co>

Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -15,7 +15,7 @@ class PreTrainedPipeline():
15
  def __init__(self, path=""):
16
  # load the optimized model
17
  self.model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth'
18
- self.model = blip_decoder(pretrained=self.model_url, image_size=384, vit='large')
19
  self.model.eval()
20
  self.model = self.model.to(device)
21
 
 
15
  def __init__(self, path=""):
16
  # load the optimized model
17
  self.model_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_large_caption.pth'
18
+ self.model = blip_decoder(pretrained=self.model_url, image_size=384, vit='large',med_config=os.path.join(path, 'configs/med_config.json'))
19
  self.model.eval()
20
  self.model = self.model.to(device)
21