locsec/AMMICO
1
Форкнуть 0
зеркало из https://github.com/ssciwr/AMMICO.git synced 2025-10-29 21:16:06 +02:00
Код Задачи Пакеты Проекты Релизы Вики Активность

Apply suggestions from code review

Просмотр исходного кода 
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Этот коммит содержится в:
Inga Ulusoy 2025-10-27 09:59:03 +01:00 коммит произвёл GitHub
родитель 237c6265fe
Коммит a65f1e2287
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 3 добавлений и 3 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

6
ammico/image_summary.py
Просмотреть файл

@ -289,7 +289,7 @@ class ImageSummaryDetector(AnalysisMethod):
max_new_tokens = self.token_prompt_config[ max_new_tokens = self.token_prompt_config[
"concise" if is_concise_summary else "default" "concise" if is_concise_summary else "default"
]["summary"]["max_new_tokens"] ]["summary"]["max_new_tokens"]
inputs = self._prepare_inputs(prompt, entry) inputs = self._prepare_inputs([prompt], entry)
gen_conf = GenerationConfig( gen_conf = GenerationConfig(
max_new_tokens=max_new_tokens, max_new_tokens=max_new_tokens,
@ -384,10 +384,10 @@ class ImageSummaryDetector(AnalysisMethod):
""" """
prompt = self.token_prompt_config[ prompt = self.token_prompt_config[
"concise" if is_concise_answer else "default" "concise" if is_concise_answer else "default"
]["answer"]["prompt"] ]["questions"]["prompt"]
max_new_tokens = self.token_prompt_config[ max_new_tokens = self.token_prompt_config[
"concise" if is_concise_answer else "default" "concise" if is_concise_answer else "default"
]["answer"]["max_new_tokens"] ]["questions"]["max_new_tokens"]
list_of_questions = self._clean_list_of_questions(list_of_questions, prompt) list_of_questions = self._clean_list_of_questions(list_of_questions, prompt)
gen_conf = GenerationConfig(max_new_tokens=max_new_tokens, do_sample=False) gen_conf = GenerationConfig(max_new_tokens=max_new_tokens, do_sample=False)