Upload TFMBartForConditionalGeneration

README.md

@@ -1,110 +1,47 @@
 ---
-datasets:
-- mlsum
-- batubayk/TR-News
-- csebuetnlp/xlsum
-- wiki_lingua
-language:
-- tr
-results:
-  - task:
-    type: text-summarization
-    dataset:
-      name: mlsum
-      type: mlsum
-    metrics:
-      - name: rogue(r1/r2/rl)
-        type: rouge
-        value: 45.75/32.71/39.86
-  - task:
-    type: text-summarization
-    dataset:
-      name: batubayk/TR-News
-      type: batubayk/TR-News
-    metrics:
-      - name: rogue(r1/r2/rl)
-        type: rouge
-        value: 41.97/28.26/36.69
-  - task:
-    type: text-summarization
-    dataset:
-      name: csebuetnlp/xlsum
-      type: csebuetnlp/xlsum
-    metrics:
-      - name: rogue(r1/r2/rl)
-        type: rouge
-        value: 34.15/17.94/28.03
-arxiv: 2403.01308
-library_name: transformers
-pipeline_tag: text2text-generation
+tags:
+- generated_from_keras_callback
+model-index:
+- name: VBART-Large-Summarization
+  results: []
 ---
-# VBART Model Card
-
-## Model Description  
-
-VBART is the first sequence-to-sequence model trained in Turkish corpora from scratch. It was developed by VNGRS in (Ne zamandı).  
-This model is capable of text transformation task such as summarization, paraphrasing, title generation with finetuning.
-
-This model is scores better on many tasks while being much smaller than other implementations.
-
-This repository contains fine-tuned weights of VBART for summarization task using Turkish sections of [mlsum](https://huggingface.co/datasets/mlsum), [TRNews](https://huggingface.co/datasets/batubayk/TR-News), [XLSum](https://huggingface.co/datasets/csebuetnlp/xlsum/viewer/turkish) and [Wikilingua](https://huggingface.co/datasets/wiki_lingua).
-
-- **Developed by:** [VNGRS](https://vngrs.com/)
-- **Model type:**  Transformer encoder-decoder based on mBart
-- **Language(s) (NLP):** Turkish
-- **License:** [More Information Needed]
-- **Finetuned from model:** VBART
-- Paper : [arxiv](https://arxiv.org/abs/2403.01308)
-## How to Get Started with the Model  
-Use the code below to get started with the model.  
--> Model yüklendikten sonra bir kod çıkar
-[More Information Needed]  
-  
-## Training Details  
-  
-### Training Data  
-Base model training data is filtered mixed corpus made of Turkish parts of [OSCAR-2201](https://huggingface.co/datasets/oscar-corpus/OSCAR-2201) and [mC4](https://huggingface.co/datasets/mc4) datasets. These datasets consist of documents of unstructured web crawl data. More information about the dataset can be found in their respective page. Data then filtered using set of heuristics and certain rules, explained in appendix of our [paper](https://arxiv.org/abs/2403.01308).
-
-Fine-tuning dataset is Turkish sections of [mlsum](https://huggingface.co/datasets/mlsum), [TRNews](https://huggingface.co/datasets/batubayk/TR-News), [XLSum](https://huggingface.co/datasets/csebuetnlp/xlsum/viewer/turkish) and [Wikilingua](https://huggingface.co/datasets/wiki_lingua), as mentioned before.
-
-
-### Limitations
-This model in fine-tuned to question answering and question generation task. It is not intended to be used in any other case and can not be fine-tuned to any other task with full performance of the base model.
-
-### Training Procedure  
-Pretrained for 30 days, resulted in total training of 23 epochs. TODO: Ne kadar token olduğunu yaz.
-#### Hardware
-- **GPUs**: 8X Nvidia A100-80 GB
-#### Software
-- Tensorflow
-#### Hyperparameters  
-##### Pretraining
-- **Training regime:** fp16 mixed precision
-- **Training objective** : Sentence permutation and span masking (using mask lengths sampled from poisson distribution $\lambda = 3.5$ and total of %30 data)
-- **Optimizer** : Adam optimizer (\(\beta_{1} = 0.9, \beta_{2} = 0.98, \epsilon = 1e-6\))
-- **Scheduler**:  Linear decay scheduler (20.000 warm up steps)
-- **Dropout**: 0.1 (dropped to 0.05 and 0 in last 160k steps)
-- **Learning rate**: \( 5e-6 \)
-##### Finetuning
-- **Training regime:** fp16 mixed precision
-- **Optimizer** : Adam optimizer (\(\beta_{1} = 0.9, \beta_{2} = 0.98, \epsilon = 1e-6\))
-- **Scheduler**:  Linear decay scheduler
-- **Dropout**: 0.1 
--  **Learning rate**: \(5e-5\)
-#### Metrics
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/62f8b3c84588fe31f435a92b/QCef-9yumzG2sHksOGcUs.png)
-
-## License
-
-
-## Citation  
-```
-@misc{VBART,
-      title={VBART: The Turkish LLM}, 
-      author={Melikşah Türker and Mehmet Erdi Arı and Aydın Han},
-      year={2024},
-      eprint={2403.01308},
-      archivePrefix={arXiv},
-      primaryClass={cs.CL}
-}
-```
+
+<!-- This model card has been generated automatically according to the information Keras had access to. You should
+probably proofread and complete it, then remove this comment. -->
+
+# VBART-Large-Summarization
+
+This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+It achieves the following results on the evaluation set:
+
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- optimizer: None
+- training_precision: float32
+
+### Training results
+
+
+
+### Framework versions
+
+- Transformers 4.38.2
+- TensorFlow 2.13.1
+- Datasets 2.18.0
+- Tokenizers 0.15.2

config.json

@@ -0,0 +1,33 @@
+{
+  "activation_dropout": 0.0,
+  "activation_function": "gelu",
+  "architectures": [
+    "MBartForConditionalGeneration"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 2,
+  "classifier_dropout": 0.0,
+  "d_model": 1024,
+  "decoder_attention_heads": 16,
+  "decoder_ffn_dim": 4096,
+  "decoder_layerdrop": 0.0,
+  "decoder_layers": 12,
+  "decoder_start_token_id": 2,
+  "dropout": 0.1,
+  "encoder_attention_heads": 16,
+  "encoder_ffn_dim": 4096,
+  "encoder_layerdrop": 0.0,
+  "encoder_layers": 12,
+  "eos_token_id": 3,
+  "forced_eos_token_id": 3,
+  "init_std": 0.02,
+  "is_encoder_decoder": true,
+  "max_position_embeddings": 1024,
+  "model_type": "mbart",
+  "num_hidden_layers": 12,
+  "pad_token_id": 0,
+  "scale_embedding": false,
+  "transformers_version": "4.38.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

generation_config.json

@@ -0,0 +1,9 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 2,
+  "decoder_start_token_id": 2,
+  "eos_token_id": 3,
+  "forced_eos_token_id": 3,
+  "pad_token_id": 0,
+  "transformers_version": "4.38.2"
+}

tf_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dba67d9e9003c7b8f0ee54bf14b6f27e37c5b05821f2fca7a5e11723afbad22e
+size 1551059288