youssef
commited on
Commit
·
3ad5e22
1
Parent(s):
c0d1640
still use sample
Browse files
src/video_processor/processor.py
CHANGED
|
@@ -104,7 +104,7 @@ Be specific about all visual elements to enable searching later."""}
|
|
| 104 |
with torch.inference_mode():
|
| 105 |
outputs = self.model.generate(
|
| 106 |
**inputs,
|
| 107 |
-
do_sample=
|
| 108 |
temperature=0.7,
|
| 109 |
max_new_tokens=256,
|
| 110 |
)
|
|
|
|
| 104 |
with torch.inference_mode():
|
| 105 |
outputs = self.model.generate(
|
| 106 |
**inputs,
|
| 107 |
+
do_sample=True,
|
| 108 |
temperature=0.7,
|
| 109 |
max_new_tokens=256,
|
| 110 |
)
|