Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|