mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
fix: only import translator when needed
This commit is contained in:
@@ -198,7 +198,6 @@ def worker():
|
|||||||
from modules.upscaler import perform_upscale
|
from modules.upscaler import perform_upscale
|
||||||
from modules.flags import Performance
|
from modules.flags import Performance
|
||||||
from modules.meta_parser import get_metadata_parser
|
from modules.meta_parser import get_metadata_parser
|
||||||
from modules.translator import translate2en
|
|
||||||
|
|
||||||
pid = os.getpid()
|
pid = os.getpid()
|
||||||
print(f'Started worker with PID {pid}')
|
print(f'Started worker with PID {pid}')
|
||||||
@@ -949,6 +948,8 @@ def worker():
|
|||||||
prompt = fallback_prompt
|
prompt = fallback_prompt
|
||||||
else:
|
else:
|
||||||
if translate:
|
if translate:
|
||||||
|
# only initialize when needed, requires an internet connection
|
||||||
|
from modules.translator import translate2en
|
||||||
prompt = translate2en(prompt, prompt_type)
|
prompt = translate2en(prompt, prompt_type)
|
||||||
return prompt
|
return prompt
|
||||||
|
|
||||||
@@ -1093,6 +1094,8 @@ def worker():
|
|||||||
set_hyper_sd_defaults(async_task, current_progress, advance_progress=True)
|
set_hyper_sd_defaults(async_task, current_progress, advance_progress=True)
|
||||||
|
|
||||||
if async_task.translate_prompts:
|
if async_task.translate_prompts:
|
||||||
|
# only initialize when needed, requires an internet connection
|
||||||
|
from modules.translator import translate2en
|
||||||
async_task.prompt = translate2en(async_task.prompt, 'prompt')
|
async_task.prompt = translate2en(async_task.prompt, 'prompt')
|
||||||
async_task.negative_prompt = translate2en(async_task.negative_prompt, 'negative prompt')
|
async_task.negative_prompt = translate2en(async_task.negative_prompt, 'negative prompt')
|
||||||
|
|
||||||
|
|||||||
@@ -1,13 +1,14 @@
|
|||||||
import translators
|
import translators
|
||||||
from functools import lru_cache
|
from functools import lru_cache
|
||||||
|
|
||||||
|
|
||||||
@lru_cache(maxsize=32, typed=False)
|
@lru_cache(maxsize=32, typed=False)
|
||||||
def translate2en(text, element):
|
def translate2en(text, element):
|
||||||
if not text:
|
if not text:
|
||||||
return text
|
return text
|
||||||
|
|
||||||
try:
|
try:
|
||||||
result = translators.translate_text(text,to_language='en')
|
result = translators.translate_text(text, to_language='en')
|
||||||
print(f'[Parameters] Translated {element}: {result}')
|
print(f'[Parameters] Translated {element}: {result}')
|
||||||
return result
|
return result
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
Reference in New Issue
Block a user