mirror of
https://github.com/Sygil-Dev/sygil-webui.git
synced 2024-12-15 06:21:34 +03:00
15 lines
352 B
YAML
15 lines
352 B
YAML
|
image_root: '/export/share/datasets/vision/nocaps/'
|
||
|
ann_root: 'annotation'
|
||
|
|
||
|
# set pretrained as a file path or an url
|
||
|
pretrained: 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_caption_capfilt_large.pth'
|
||
|
|
||
|
vit: 'base'
|
||
|
batch_size: 32
|
||
|
|
||
|
image_size: 384
|
||
|
|
||
|
max_length: 20
|
||
|
min_length: 5
|
||
|
num_beams: 3
|
||
|
prompt: 'a picture of '
|