Upload generate_prompt.py with huggingface_hub

generate_prompt.py

@@ -87,9 +87,9 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
 
         data_list.append(messages)
         save_list.append(save_)
-    print(len(data_list))
+    #print(len(data_list))
     text = processor.apply_chat_template(data_list, tokenize=False, add_generation_prompt=True)
-    print(len(text))
+    #print(len(text))
     image_inputs, video_inputs = process_vision_info(data_list)
     inputs = processor(
         text=[text],
@@ -102,7 +102,7 @@ for batch_idx in tqdm(range(begin, end, batch_size)):
 
     # Inference: Generation of the output
     generated_ids = model.generate(**inputs, max_new_tokens=128)
-    print(generated_ids.shape)
+    #print(generated_ids.shape)
     generated_ids_trimmed = [
         out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
     ]