anas-awadalla commited on
Commit
d2854f1
·
1 Parent(s): 973d62e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -157,7 +157,7 @@ def generate(
157
  lang_x=input_ids,
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
- num_beams=1,
161
  do_sample=True,
162
  temperature=0.3,
163
  top_k=0,
@@ -202,7 +202,7 @@ with gr.Blocks() as demo:
202
 
203
  GitHub: [open_flamingo](https://github.com/mlfoundations/open_flamingo)
204
 
205
- In this demo we showcase the in-context learning capabilities of the OpenFlamingo-4B model, a large multimodal model trained on top of RedPajama-INCITE-Instruct-3B-v1.
206
  The model is trained on an interleaved mixture of text and images and is able to generate text conditioned on sequences of images/text. To safeguard against harmful generations, we detect toxic text in the model output and reject it. However, we understand that this is not a perfect solution and we encourage you to use this demo responsibly. If you find that the model is generating harmful text, please report it using this [form](https://forms.gle/StbcPvyyW2p3Pc7z6).
207
  """
208
  )
 
157
  lang_x=input_ids,
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
+ num_beams=3,
161
  do_sample=True,
162
  temperature=0.3,
163
  top_k=0,
 
202
 
203
  GitHub: [open_flamingo](https://github.com/mlfoundations/open_flamingo)
204
 
205
+ In this demo we showcase the in-context learning capabilities of the OpenFlamingo-4B model, a large multimodal model trained on top of RedPajama-INCITE-Instruct-3B-v1. Note that we add two additional demonstrations to the ones presented to improve the demo experience.
206
  The model is trained on an interleaved mixture of text and images and is able to generate text conditioned on sequences of images/text. To safeguard against harmful generations, we detect toxic text in the model output and reject it. However, we understand that this is not a perfect solution and we encourage you to use this demo responsibly. If you find that the model is generating harmful text, please report it using this [form](https://forms.gle/StbcPvyyW2p3Pc7z6).
207
  """
208
  )