alignment-handbook icon indicating copy to clipboard operation
alignment-handbook copied to clipboard

ModuleNotFoundError: No module named 'h4'

Open AIR-hl opened this issue 6 months ago • 0 comments

Hello, I want to load the training_arg.bin of zephyr-7b-beta and pass it to DPOTrainer of trl to compute the implicit rewards and logps conveniently, but it seems lack some private modules.

import torch
training_args = torch.load("training_args.bin")

ModuleNotFoundError: No module named 'h4'

AIR-hl avatar Aug 15 '24 05:08 AIR-hl