pietre/rss2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Preparar repositorio para despliegue: código fuente limpio

Browse source
This commit is contained in:
jlimolina 2026-01-23 02:00:40 +01:00
parent 866f5c432d
commit 3eca832c1a
76 changed files with 5434 additions and 3496 deletions
Download patch file Download diff file Expand all files Collapse all files

4
workers/translation_worker.py
View file

@ -73,7 +73,7 @@ MAX_NEW_TOKENS_TITLE = _env_int("MAX_NEW_TOKENS_TITLE", 96)
MAX_NEW_TOKENS_BODY = _env_int("MAX_NEW_TOKENS_BODY", 512)
NUM_BEAMS_TITLE = _env_int("NUM_BEAMS_TITLE", 2)
NUM_BEAMS_BODY = _env_int("NUM_BEAMS_BODY", 1)
NUM_BEAMS_BODY = _env_int("NUM_BEAMS_BODY", 2)
# HuggingFace model name (used for tokenizer)
UNIVERSAL_MODEL = _env_str("UNIVERSAL_MODEL", "facebook/nllb-200-distilled-600M")
@ -304,6 +304,8 @@ def _translate_texts(src, tgt, texts, beams, max_new_tokens):
target_prefix=target_prefix,
beam_size=beams,
max_decoding_length=max_new,
repetition_penalty=1.1,
no_repeat_ngram_size=4,
)
dt = time.time() - start