sivan22
/

hdd-words-ocr

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

sivan22 commited on May 17, 2023

Commit

3947bd0

•

1 Parent(s): 6bc2b0f

Upload config

Files changed (1) hide show

config.json +1 -5

config.json CHANGED Viewed

@@ -1,8 +1,5 @@
 {
   "_commit_hash": null,
-  "architectures": [
-    "VisionEncoderDecoderModel"
-  ],
   "decoder": {
     "_name_or_path": "imvladikon/alephbertgimmel-base-512",
     "add_cross_attention": true,
@@ -170,13 +167,12 @@
   "eos_token_id": 2,
   "is_encoder_decoder": true,
   "length_penalty": 2.0,
-  "max_length": 5,
   "model_type": "vision-encoder-decoder",
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
   "pad_token_id": 3,
   "tie_word_embeddings": false,
-  "torch_dtype": "float32",
   "transformers_version": null,
   "vocab_size": 128000
 }

 {
   "_commit_hash": null,
   "decoder": {
     "_name_or_path": "imvladikon/alephbertgimmel-base-512",
     "add_cross_attention": true,
   "eos_token_id": 2,
   "is_encoder_decoder": true,
   "length_penalty": 2.0,
+  "max_length": 4,
   "model_type": "vision-encoder-decoder",
   "no_repeat_ngram_size": 3,
   "num_beams": 4,
   "pad_token_id": 3,
   "tie_word_embeddings": false,
   "transformers_version": null,
   "vocab_size": 128000
 }