chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-04 18:10:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

set num_beam default to 1 instead of completely removing it

Browse Source
This commit is contained in:
152334H 2023-04-17 14:56:18 +08:00
parent 6a8e6e49ff
commit 700f05d079
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo.py
View File

@ -113,8 +113,8 @@ with gr.Blocks() as demo:
num_beams = gr.Slider(
minimum=1,
maximum=16,
value=5,
maximum=10,
value=1,
step=1,
interactive=True,
label="beam search numbers)",

4
minigpt4/conversation/conversation.py
View File

@ -134,7 +134,7 @@ class Chat:
else:
conv.append_message(conv.roles[0], text)
def answer(self, conv, img_list, max_new_tokens=200, num_beams=5, min_length=1, top_p=0.9,
def answer(self, conv, img_list, max_new_tokens=200, num_beams=1, min_length=1, top_p=0.9,
repetition_penalty=1.0, length_penalty=1, temperature=1.0):
conv.append_message(conv.roles[1], None)
embs = self.get_context_emb(conv, img_list)
@ -142,7 +142,7 @@ class Chat:
inputs_embeds=embs,
max_new_tokens=max_new_tokens,
stopping_criteria=self.stopping_criteria,
#num_beams=num_beams,
num_beams=num_beams,
do_sample=True,
min_length=min_length,
top_p=top_p,