Commit
•
da8e8bd
1
Parent(s):
6192f9b
Upload model
Browse files- config.json +2 -1
- model.safetensors +1 -1
config.json
CHANGED
|
@@ -95,6 +95,7 @@
|
|
| 95 |
"vocab_size": 50257
|
| 96 |
},
|
| 97 |
"decoder_start_token_id": 5,
|
|
|
|
| 98 |
"encoder": {
|
| 99 |
"_name_or_path": "microsoft/git-large-coco",
|
| 100 |
"add_cross_attention": false,
|
|
@@ -255,7 +256,7 @@
|
|
| 255 |
"eos_token_id": 6,
|
| 256 |
"is_encoder_decoder": true,
|
| 257 |
"model_type": "vision-encoder-decoder",
|
| 258 |
-
"pad_token_id":
|
| 259 |
"tie_word_embeddings": false,
|
| 260 |
"torch_dtype": "float32",
|
| 261 |
"transformers_version": "4.40.0",
|
|
|
|
| 95 |
"vocab_size": 50257
|
| 96 |
},
|
| 97 |
"decoder_start_token_id": 5,
|
| 98 |
+
"device_map": "cuda:0",
|
| 99 |
"encoder": {
|
| 100 |
"_name_or_path": "microsoft/git-large-coco",
|
| 101 |
"add_cross_attention": false,
|
|
|
|
| 256 |
"eos_token_id": 6,
|
| 257 |
"is_encoder_decoder": true,
|
| 258 |
"model_type": "vision-encoder-decoder",
|
| 259 |
+
"pad_token_id": 3,
|
| 260 |
"tie_word_embeddings": false,
|
| 261 |
"torch_dtype": "float32",
|
| 262 |
"transformers_version": "4.40.0",
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 2094222336
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f2232136289813a316bf7a167f006c0e1c257e8b54e1542285c2647731230879
|
| 3 |
size 2094222336
|