From 25ee5ba1e48d0e0f29f65d826579533540bc7fb0 Mon Sep 17 00:00:00 2001 From: TAE-IN Sappho KIM Date: Thu, 13 Jun 2024 13:02:47 +0900 Subject: [PATCH] =?UTF-8?q?Rename=20FF14Translator=E2=86=92FF14JaKoTransla?= =?UTF-8?q?tor?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 12 ++++++------ .../{FF14Translator.cs => FF14JaKoTranslator.cs} | 4 ++-- src/ExampleApp/Program.cs | 6 +++--- 3 files changed, 11 insertions(+), 11 deletions(-) rename src/EDMTranslator/Translation/{FF14Translator.cs => FF14JaKoTranslator.cs} (98%) diff --git a/README.md b/README.md index fad2605..b2f2912 100644 --- a/README.md +++ b/README.md @@ -26,12 +26,12 @@ Text translator library based on LLM models, especially EncoderDecoderModel in H # Supported models -* FF14Translator([sappho192/ffxiv-ja-ko-translator](https://github.com/sappho192/ffxiv-ja-ko-translator)): Japanese-to-Korean translator based on `bert-base-japanese` and `skt-kogpt2-base-v2` +* FF14JaKoTranslator([sappho192/ffxiv-ja-ko-translator](https://github.com/sappho192/ffxiv-ja-ko-translator)): Japanese-to-Korean translator based on `bert-base-japanese` and `skt-kogpt2-base-v2` * More to be added... # Quickstart -Following guide supposes that you are to use FF14Translator mentioned above. +Following guide supposes that you are to use FF14JaKoTranslator mentioned above. ## Install the packages @@ -54,7 +54,7 @@ Write the code like below and you are good to go 🫑 Note that you need to fix the path of `encoderDictDir` and `modelDir` correctly. ```csharp - // Console application which translates Japanese sentence to Korean based on FF14Translator + // Console application which translates Japanese sentence to Korean based on FF14JaKoTranslator using EDMTranslator.Tokenization; using EDMTranslator.Translation; @@ -90,8 +90,8 @@ TestTokenizer(tokenizer); // Prepare the translator string modelDir = @"D:\MODEL\ffxiv-ja-ko-translator\onnx"; // Contains encoder_model.onnx and decoder_model_merged.onnx -var translator = new FF14Translator(tokenizer, modelDir); -void TestTranslator(FF14Translator translator) +var translator = new FF14JaKoTranslator(tokenizer, modelDir); +void TestTranslator(FF14JaKoTranslator translator) { Console.WriteLine("--Translator test--"); Translate(translator, "ζ‰“γ‘εˆγ‚γ›γŒη΅‚γ‚γ£γŸεΎŒγ«γ”ι£―γ‚’ι£ŸγΉγΎγ—γ‚‡γ†γ€‚"); @@ -101,7 +101,7 @@ void TestTranslator(FF14Translator translator) } TestTranslator(translator); -static void Translate(FF14Translator translator, string sentence) +static void Translate(FF14JaKoTranslator translator, string sentence) { Console.WriteLine($"SourceText: {sentence}"); string translated = translator.Translate(sentence); diff --git a/src/EDMTranslator/Translation/FF14Translator.cs b/src/EDMTranslator/Translation/FF14JaKoTranslator.cs similarity index 98% rename from src/EDMTranslator/Translation/FF14Translator.cs rename to src/EDMTranslator/Translation/FF14JaKoTranslator.cs index 88994c6..c66f38d 100644 --- a/src/EDMTranslator/Translation/FF14Translator.cs +++ b/src/EDMTranslator/Translation/FF14JaKoTranslator.cs @@ -6,12 +6,12 @@ namespace EDMTranslator.Translation { - public class FF14Translator : EncoderDecoderTranslator + public class FF14JaKoTranslator : EncoderDecoderTranslator { private readonly InferenceSession encoderSession; private readonly InferenceSession decoderSession; - public FF14Translator(BertJa2KoGPTTokenizer tokenizer, string modelDirPath) : base( + public FF14JaKoTranslator(BertJa2KoGPTTokenizer tokenizer, string modelDirPath) : base( modelName: "ffxiv-ja-ko-translator", modelHubName: "sappho192/ffxiv-ja-ko-translator", tokenizer: tokenizer) diff --git a/src/ExampleApp/Program.cs b/src/ExampleApp/Program.cs index 98c6daf..ee80ee1 100644 --- a/src/ExampleApp/Program.cs +++ b/src/ExampleApp/Program.cs @@ -32,8 +32,8 @@ void TestTokenizer(ITokenizer tokenizer) // Prepare the translator string modelDir = @"D:\MODEL\ffxiv-ja-ko-translator\onnx"; -var translator = new FF14Translator(tokenizer, modelDir); -void TestTranslator(FF14Translator translator) +var translator = new FF14JaKoTranslator(tokenizer, modelDir); +void TestTranslator(FF14JaKoTranslator translator) { Console.WriteLine("--Translator test--"); Translate(translator, "ζ‰“γ‘εˆγ‚γ›γŒη΅‚γ‚γ£γŸεΎŒγ«γ”ι£―γ‚’ι£ŸγΉγΎγ—γ‚‡γ†γ€‚"); @@ -43,7 +43,7 @@ void TestTranslator(FF14Translator translator) } TestTranslator(translator); -static void Translate(FF14Translator translator, string sentence) +static void Translate(FF14JaKoTranslator translator, string sentence) { Console.WriteLine($"SourceText: {sentence}"); string translated = translator.Translate(sentence);