lakshayt commited on
Commit
8e600bb
1 Parent(s): 188ba58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,8 +10,7 @@ model, image_processor, tokenizer = create_model_and_transforms(
10
  clip_vision_encoder_pretrained="openai",
11
  lang_encoder_path="anas-awadalla/mpt-1b-redpajama-200b",
12
  tokenizer_path="anas-awadalla/mpt-1b-redpajama-200b",
13
- cross_attn_every_n_layers=1,
14
- cache_dir="PATH/TO/CACHE/DIR" # Defaults to ~/.cache
15
  )
16
 
17
  # grab model checkpoint from huggingface hub
 
10
  clip_vision_encoder_pretrained="openai",
11
  lang_encoder_path="anas-awadalla/mpt-1b-redpajama-200b",
12
  tokenizer_path="anas-awadalla/mpt-1b-redpajama-200b",
13
+ cross_attn_every_n_layers=1
 
14
  )
15
 
16
  # grab model checkpoint from huggingface hub