RamAnanth1 commited on
Commit
9b9d533
1 Parent(s): 887b324

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -30,8 +30,7 @@ def infer(image, prompt, min_len, max_len, beam_size, len_penalty, repetition_pe
30
  max_length=max_len,
31
  min_length=min_len,
32
  top_p=top_p,
33
- use_nucleus_sampling=use_nucleus_sampling,
34
- dtype=torch.float16
35
  )
36
 
37
  return output[0]
 
30
  max_length=max_len,
31
  min_length=min_len,
32
  top_p=top_p,
33
+ use_nucleus_sampling=use_nucleus_sampling
 
34
  )
35
 
36
  return output[0]