diff --git a/.gitignore b/.gitignore index 854140d..40f05b4 100644 --- a/.gitignore +++ b/.gitignore @@ -1,5 +1,5 @@ .venv/ llama.cpp/ -datasets/ +NEREL/ __pycache__/ models/ diff --git a/modules/paraGenerator.py b/modules/paraGenerator.py index ddd5af2..d8fbfc9 100644 --- a/modules/paraGenerator.py +++ b/modules/paraGenerator.py @@ -9,7 +9,7 @@ class ParaphraseGenerator: Использование: ParaphraseGenerator([температура], [максимальное количество токенов]) """ - def __init__(self, temperature=0.8, max_tokens=512): + def __init__(self, temperature=0.8, max_tokens=2048): self.client = OpenAI(base_url='http://127.0.0.1:8080', api_key='') self.temperature = temperature self.max_tokens = max_tokens