fix some errors due to inconsistency of model cards
This commit is contained in:
1 parent
9ac0fde3f2
commit
2166b9bc6b
5 files changed
+66
-69
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user