Upload model

config.json

@@ -4,7 +4,6 @@
     "EncoderDecoderModel"
   ],
   "decoder": {
-    "_attn_implementation_autoset": false,
     "_name_or_path": "seyonec/ChemBERTa-zinc-base-v1",
     "add_cross_attention": true,
     "architectures": [
@@ -86,9 +85,8 @@
     "vocab_size": 767
   },
   "decoder_start_token_id": 0,
-  "do_sample": null,
+  "do_sample": true,
   "encoder": {
-    "_attn_implementation_autoset": false,
     "_name_or_path": "seyonec/ChemBERTa-zinc-base-v1",
     "add_cross_attention": false,
     "architectures": [
@@ -171,12 +169,12 @@
   },
   "eos_token_id": 2,
   "is_encoder_decoder": true,
-  "max_length": null,
+  "max_length": 512,
   "model_type": "encoder-decoder",
-  "num_beams": null,
+  "num_beams": 5,
   "pad_token_id": 1,
-  "top_k": null,
+  "top_k": 20,
   "torch_dtype": "float32",
-  "transformers_version": "4.47.1",
+  "transformers_version": "4.44.2",
   "vocab_size": 767
 }

generation_config.json

@@ -6,5 +6,5 @@
   "num_beams": 5,
   "pad_token_id": 1,
   "top_k": 20,
-  "transformers_version": "4.47.1"
+  "transformers_version": "4.44.2"
 }