every_holiday/VRCT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛[bugfix] Model : Cranslate2の不要なコードを削除

Browse Source
This commit is contained in:
misyaguziya
2024-01-10 15:37:56 +09:00
parent 2aaae11e97
commit 5f1eb70516
2 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models/translation/translation_translator.py
View File

@@ -6,11 +6,6 @@ from .translation_languages import translation_lang
import ctranslate2 import ctranslate2
import transformers import transformers
TRANSLATE_MODELS = {
"small": "facebook/m2m100_418M",
"large": "facebook/m2m100_1.2B"
}
# Translator # Translator
class Translator(): class Translator():
def __init__(self, path, weight_config): def __init__(self, path, weight_config):

2
models/translation/utils.py
View File

@@ -2,7 +2,7 @@ import tempfile
from zipfile import ZipFile from zipfile import ZipFile
from os import path as os_path from os import path as os_path
from os import makedirs as os_makedirs 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 tqdm import tqdm
from typing import Callable from typing import Callable
import hashlib import hashlib