Further fix for MiniCPM-V-2_6 example (#11965)

This commit is contained in:
Yuwen Hu 2024-08-29 19:14:13 +08:00 committed by GitHub
parent 431affd0a0
commit 2e49e1f8e9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -71,7 +71,6 @@ if __name__ == '__main__':
model.chat(
image=None,
msgs=msgs,
context=None,
tokenizer=tokenizer,
)
@ -79,7 +78,6 @@ if __name__ == '__main__':
res = model.chat(
image=None,
msgs=msgs,
context=None,
tokenizer=tokenizer,
stream=True
)
@ -96,7 +94,6 @@ if __name__ == '__main__':
res = model.chat(
image=None,
msgs=msgs,
context=None,
tokenizer=tokenizer,
)
torch.xpu.synchronize()