Dileep7729
commited on
Commit
•
f1cb40e
1
Parent(s):
862807d
Upload 5 files
Browse filesThis model is a fine-tuned version of Salesforce/blip-image-captioning-base on a custom dataset.
- config.json +29 -0
- generation_config.json +7 -0
- model.safetensors +3 -0
- preprocessor_config.json +17 -0
- training_args.bin +3 -0
config.json
ADDED
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "Salesforce/blip-image-captioning-base",
|
3 |
+
"architectures": [
|
4 |
+
"BlipForConditionalGeneration"
|
5 |
+
],
|
6 |
+
"image_text_hidden_size": 256,
|
7 |
+
"initializer_factor": 1.0,
|
8 |
+
"initializer_range": 0.02,
|
9 |
+
"label_smoothing": 0.0,
|
10 |
+
"logit_scale_init_value": 2.6592,
|
11 |
+
"model_type": "blip",
|
12 |
+
"projection_dim": 512,
|
13 |
+
"text_config": {
|
14 |
+
"_attn_implementation_autoset": true,
|
15 |
+
"initializer_factor": 1.0,
|
16 |
+
"model_type": "blip_text_model",
|
17 |
+
"num_attention_heads": 12
|
18 |
+
},
|
19 |
+
"torch_dtype": "float32",
|
20 |
+
"transformers_version": "4.46.1",
|
21 |
+
"vision_config": {
|
22 |
+
"_attn_implementation_autoset": true,
|
23 |
+
"dropout": 0.0,
|
24 |
+
"initializer_factor": 1.0,
|
25 |
+
"initializer_range": 0.02,
|
26 |
+
"model_type": "blip_vision_model",
|
27 |
+
"num_channels": 3
|
28 |
+
}
|
29 |
+
}
|
generation_config.json
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_from_model_config": true,
|
3 |
+
"bos_token_id": 30522,
|
4 |
+
"eos_token_id": 2,
|
5 |
+
"pad_token_id": 0,
|
6 |
+
"transformers_version": "4.46.1"
|
7 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4465cbd71678ee874cff884a282e2144bf56e85e8175574305365911f6456876
|
3 |
+
size 989717056
|
preprocessor_config.json
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"do_normalize": true,
|
3 |
+
"do_resize": true,
|
4 |
+
"image_mean": [
|
5 |
+
0.48145466,
|
6 |
+
0.4578275,
|
7 |
+
0.40821073
|
8 |
+
],
|
9 |
+
"image_processor_type": "BlipImageProcessor",
|
10 |
+
"image_std": [
|
11 |
+
0.26862954,
|
12 |
+
0.26130258,
|
13 |
+
0.27577711
|
14 |
+
],
|
15 |
+
"processor_class": "BlipProcessor",
|
16 |
+
"size": 384
|
17 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:462d7f4d862fbaa5bb5f6c91c0b40a0276f2a1d15272e24f39357f5b50f7940e
|
3 |
+
size 5240
|