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
|