salesforce/BLIP
1
0
Fork 0
mirror of https://github.com/salesforce/BLIP.git synced 2026-04-30 12:21:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

update demo

Browse Source
This commit is contained in:
root
2022-02-11 07:55:17 +00:00
parent a3a2577a20
commit ad5eec314c
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
demo.ipynb
View File

@@ -106,7 +106,10 @@
"model = model.to(device)\n", "model = model.to(device)\n",
"\n", "\n",
"with torch.no_grad():\n", "with torch.no_grad():\n",
" caption = model.generate(image, sample=False, num_beams=3, max_length=20, min_length=5)\n", " # beam search\n",
" caption = model.generate(image, sample=False, num_beams=3, max_length=20, min_length=5) \n",
" # nucleus sampling\n",
" # caption = model.generate(image, sample=True, top_p=0.9, max_length=20, min_length=5) \n",
" print('caption: '+caption[0])" " print('caption: '+caption[0])"
] ]
}, },