From 5f1eb7051617db6f0838e47e0844fd2cac636fd6 Mon Sep 17 00:00:00 2001 From: misyaguziya Date: Wed, 10 Jan 2024 15:37:56 +0900 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B[bugfix]=20Model=20:=20Cranslate2?= =?UTF-8?q?=E3=81=AE=E4=B8=8D=E8=A6=81=E3=81=AA=E3=82=B3=E3=83=BC=E3=83=89?= =?UTF-8?q?=E3=82=92=E5=89=8A=E9=99=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- models/translation/translation_translator.py | 5 ----- models/translation/utils.py | 2 +- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/models/translation/translation_translator.py b/models/translation/translation_translator.py index 21710e2d..8c3e401b 100644 --- a/models/translation/translation_translator.py +++ b/models/translation/translation_translator.py @@ -6,11 +6,6 @@ from .translation_languages import translation_lang import ctranslate2 import transformers -TRANSLATE_MODELS = { - "small": "facebook/m2m100_418M", - "large": "facebook/m2m100_1.2B" -} - # Translator class Translator(): def __init__(self, path, weight_config): diff --git a/models/translation/utils.py b/models/translation/utils.py index c41fbb88..f1c40b9e 100644 --- a/models/translation/utils.py +++ b/models/translation/utils.py @@ -2,7 +2,7 @@ import tempfile from zipfile import ZipFile from os import path as os_path from os import makedirs as os_makedirs -from requests import get as requests_get, head as requests_head +from requests import get as requests_get from tqdm import tqdm from typing import Callable import hashlib