mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-23 14:22:51 +08:00
fix paligemma infer
Former-commit-id: 84cd1188ac03c165e1a626db297936c2458627d6
This commit is contained in:
parent
9d437a5f4f
commit
f07bad7144
@ -171,7 +171,7 @@ class HuggingfaceEngine(BaseEngine):
|
||||
elif not isinstance(value, torch.Tensor):
|
||||
value = torch.tensor(value)
|
||||
|
||||
gen_kwargs[key] = value.to(model.device)
|
||||
gen_kwargs[key] = value.to(dtype=model.dtype, device=model.device)
|
||||
|
||||
return gen_kwargs, prompt_length
|
||||
|
||||
|
@ -1119,7 +1119,7 @@ LOCALES = {
|
||||
"info": "Нормализация оценок в тренировке PPO.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "奖励模型",
|
||||
"label": "归一化分数",
|
||||
"info": "PPO 训练中归一化奖励分数。",
|
||||
},
|
||||
"ko": {
|
||||
|
Loading…
x
Reference in New Issue
Block a user