Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
•
9b9d533
1
Parent(s):
887b324
Update app.py
Browse files
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]
|