Feature Extraction
Transformers
Safetensors
vision-encoder-decoder
custom_code
anicolson commited on
Commit
5a67bca
1 Parent(s): c0566dd

Upload model

Browse files
Files changed (3) hide show
  1. config.json +2 -1
  2. generation_config.json +1 -1
  3. 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.40.2"
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.40.2"
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