fix some errors due to inconsistency of model cards

This commit is contained in:
Kingsley committed 2024-09-30 19:58:34 +08:00
1 parent 9ac0fde3f2
commit 2166b9bc6b
5 files changed
+66 -69

No files matched your search

+1 -1
View File
@@ -168,7 +168,7 @@ class HuggingfaceEngine(BaseEngine):
for key, value in mm_inputs.items():
value = value if isinstance(value, torch.Tensor) else torch.tensor(value)
gen_kwargs[key] = value.to(model.device)
return gen_kwargs, prompt_length
@staticmethod