mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-10-14 15:52:49 +08:00
fix paligemma infer
Former-commit-id: d272455d6118c1d670c70cfe3458d8dab111da6c
This commit is contained in:
parent
eba78f2159
commit
adff887659
@ -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