File size: 695 Bytes
a43ef32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
model:
  vit_model: "clip_L"

  qformer_num_query_token: 16
  qformer_cross_attention_freq: 1

  sd_train_text_encoder: False
  sd_pretrained_model_name_or_path: "runwayml/stable-diffusion-v1-5" 

  load_finetuned: False
  load_pretrained: True
  # pretrained: "/export/share/dongxuli/zerobooth/500000-renamed/"
  pretrained: "https://storage.googleapis.com/sfr-vision-language-research/LAVIS/models/BLIP-Diffusion/blip-diffusion.tar.gz"

preprocess:
  vis_processor:
    train:
      name: "blip_diffusion_inp_image_eval"
    eval:
      name: "blip_diffusion_inp_image_eval"
  text_processor:
    train:
      name: "blip_caption"
    eval:
      name: "blip_caption"