Solo448 commited on
Commit
9f1cd5e
·
verified ·
1 Parent(s): 4be9788

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,8 +61,8 @@ def get_qwen_op(image_file, model, processor):
61
  inputs = {k: v.to(torch.float32) if torch.is_floating_point(v) else v for k, v in inputs.items()}
62
 
63
  generation_config = {
64
- "max_new_tokens": 100000,
65
- "do_sample": True,
66
  "top_k": 20,
67
  "top_p": 0.90,
68
  "temperature": 0.4,
 
61
  inputs = {k: v.to(torch.float32) if torch.is_floating_point(v) else v for k, v in inputs.items()}
62
 
63
  generation_config = {
64
+ "max_new_tokens": 10000,
65
+ "do_sample": False,
66
  "top_k": 20,
67
  "top_p": 0.90,
68
  "temperature": 0.4,