Improve model card with pipeline tag and library name
#1
by
nielsr HF Staff - opened
README.md
CHANGED
|
@@ -1,7 +1,9 @@
|
|
| 1 |
---
|
| 2 |
-
license: mit
|
| 3 |
base_model:
|
| 4 |
- lmms-lab/LLaVA-Video-7B-Qwen2
|
|
|
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
# LLaVA-Video-7B-Qwen2-UnifiedReward-DPO
|
|
@@ -65,7 +67,8 @@ video,frame_time,video_time = load_video(video_path, max_frames_num, 1, force_sa
|
|
| 65 |
video = image_processor.preprocess(video, return_tensors="pt")["pixel_values"].cuda().half()
|
| 66 |
video = [video]
|
| 67 |
conv_template = "qwen_1_5" # Make sure you use correct chat template for different models
|
| 68 |
-
question = DEFAULT_IMAGE_TOKEN + "
|
|
|
|
| 69 |
conv = copy.deepcopy(conv_templates[conv_template])
|
| 70 |
conv.append_message(conv.roles[0], question)
|
| 71 |
conv.append_message(conv.roles[1], None)
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
base_model:
|
| 3 |
- lmms-lab/LLaVA-Video-7B-Qwen2
|
| 4 |
+
license: mit
|
| 5 |
+
pipeline_tag: video-text-to-text
|
| 6 |
+
library_name: llava
|
| 7 |
---
|
| 8 |
|
| 9 |
# LLaVA-Video-7B-Qwen2-UnifiedReward-DPO
|
|
|
|
| 67 |
video = image_processor.preprocess(video, return_tensors="pt")["pixel_values"].cuda().half()
|
| 68 |
video = [video]
|
| 69 |
conv_template = "qwen_1_5" # Make sure you use correct chat template for different models
|
| 70 |
+
question = DEFAULT_IMAGE_TOKEN + "
|
| 71 |
+
Please describe this video in detail."
|
| 72 |
conv = copy.deepcopy(conv_templates[conv_template])
|
| 73 |
conv.append_message(conv.roles[0], question)
|
| 74 |
conv.append_message(conv.roles[1], None)
|