Spaces:
Runtime error
Runtime error
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]
|