Upload model
Browse files- config.json +2 -1
- generation_config.json +1 -1
- modelling_uniformer.py +1 -0
config.json
CHANGED
@@ -49,6 +49,7 @@
|
|
49 |
"max_length": 20,
|
50 |
"max_position_embeddings": 2048,
|
51 |
"min_length": 0,
|
|
|
52 |
"model_type": "llama",
|
53 |
"no_repeat_ngram_size": 0,
|
54 |
"num_attention_heads": 12,
|
@@ -217,5 +218,5 @@
|
|
217 |
"model_type": "vision-encoder-decoder",
|
218 |
"tie_word_embeddings": false,
|
219 |
"torch_dtype": "float32",
|
220 |
-
"transformers_version": "4.
|
221 |
}
|
|
|
49 |
"max_length": 20,
|
50 |
"max_position_embeddings": 2048,
|
51 |
"min_length": 0,
|
52 |
+
"mlp_bias": false,
|
53 |
"model_type": "llama",
|
54 |
"no_repeat_ngram_size": 0,
|
55 |
"num_attention_heads": 12,
|
|
|
218 |
"model_type": "vision-encoder-decoder",
|
219 |
"tie_word_embeddings": false,
|
220 |
"torch_dtype": "float32",
|
221 |
+
"transformers_version": "4.42.3"
|
222 |
}
|
generation_config.json
CHANGED
@@ -3,5 +3,5 @@
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 4,
|
6 |
-
"transformers_version": "4.
|
7 |
}
|
|
|
3 |
"bos_token_id": 1,
|
4 |
"eos_token_id": 2,
|
5 |
"pad_token_id": 4,
|
6 |
+
"transformers_version": "4.42.3"
|
7 |
}
|
modelling_uniformer.py
CHANGED
@@ -373,6 +373,7 @@ class MultiUniFormerWithProjectionHead(UniFormerPreTrainedModel):
|
|
373 |
pixel_values: Optional[torch.Tensor] = None,
|
374 |
output_hidden_states: Optional[bool] = None,
|
375 |
return_dict: Optional[bool] = None,
|
|
|
376 |
) -> Union[Tuple, ModelOutput]:
|
377 |
|
378 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
|
|
373 |
pixel_values: Optional[torch.Tensor] = None,
|
374 |
output_hidden_states: Optional[bool] = None,
|
375 |
return_dict: Optional[bool] = None,
|
376 |
+
output_attentions: Optional[bool] = None,
|
377 |
) -> Union[Tuple, ModelOutput]:
|
378 |
|
379 |
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|