From 17d344e17ce2b9c25c0e68c1a0d69e67d965fae1 Mon Sep 17 00:00:00 2001 From: Hongji Zhu Date: Tue, 6 Aug 2024 18:06:58 +0800 Subject: [PATCH] Update README.md --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 7728c2c..d9017e1 100644 --- a/README.md +++ b/README.md @@ -1423,7 +1423,7 @@ model = AutoModel.from_pretrained('openbmb/MiniCPM-V-2_6', trust_remote_code=Tru model = model.eval().cuda() tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-V-2_6', trust_remote_code=True) -MAX_NUM_FRAMES=64 +MAX_NUM_FRAMES=64 # if cuda OOM set a smaller number def encode_video(video_path): def uniform_sample(l, n): @@ -1449,6 +1449,7 @@ msgs = [ ] # Set decode params for video +params = {} params["use_image_id"] = False params["max_slice_nums"] = 2 # use 1 if cuda OOM and video resolution > 448*448