johann22 commited on
Commit
2589a5c
·
1 Parent(s): 9425383

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -105,17 +105,17 @@ def run_idefics(in_prompt,history,image,model_drop,seed):
105
  print (seed)
106
  generate_kwargs = dict(
107
  temperature=1.0,
108
- max_new_tokens=1048,
109
  top_p=0.99,
110
  repetition_penalty=1.0,
111
  do_sample=True,
112
  seed=seed,
113
  )
114
  content = f'{agent.IDEFICS_PROMPT}' + prompt +f'![]({image})'
115
- send_list.append(agent.IDEFICS_PROMPT)
116
- send_list.append(prompt)
117
- send_list.append(image)
118
- #print(content)
119
  stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=False)
120
  resp = ""
121
  for response in stream:
 
105
  print (seed)
106
  generate_kwargs = dict(
107
  temperature=1.0,
108
+ max_new_tokens=4096,
109
  top_p=0.99,
110
  repetition_penalty=1.0,
111
  do_sample=True,
112
  seed=seed,
113
  )
114
  content = f'{agent.IDEFICS_PROMPT}' + prompt +f'![]({image})'
115
+ #send_list.append(agent.IDEFICS_PROMPT)
116
+ #send_list.append(prompt)
117
+ #send_list.append(image)
118
+ print(content)
119
  stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=False)
120
  resp = ""
121
  for response in stream: