From 83939033d13c0b8abcf67a656e132b69e86b6f4e Mon Sep 17 00:00:00 2001 From: kushal-10 Date: Sun, 5 May 2024 21:56:39 +0200 Subject: [PATCH] rm idefics log --- backends/huggingface_multimodal_api.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/backends/huggingface_multimodal_api.py b/backends/huggingface_multimodal_api.py index 59ca157f36..babc05a2f3 100644 --- a/backends/huggingface_multimodal_api.py +++ b/backends/huggingface_multimodal_api.py @@ -177,7 +177,6 @@ def generate_idefics_output(messages: list[Dict], idefics_input.append('\nAssistant:') idefics_input = [idefics_input] - print(f"Idefics input - {idefics_input}\n") inputs = processor(idefics_input, add_end_of_utterance_token=False, return_tensors="pt").to(device) # Generation args for Idefics @@ -187,7 +186,6 @@ def generate_idefics_output(messages: list[Dict], max_tokens = 2048 # Default value for input max length = 20, set a high value for now generated_ids = model.generate(**inputs, eos_token_id=exit_condition, bad_words_ids=bad_words_ids, max_length=max_tokens) generated_text = processor.batch_decode(generated_ids) - print(f"Idefics Output : {generated_text}\n") return generated_text