not found utils.py in reporsitory
koke2c95 opened this issue · 2 comments
koke2c95 commented
from utils import generate, generate_and_return_eos_logprob
need utils.py for running rng demo. also can you attach these peft ckpt (rng, arithmetic, infilling, next sentence) ? thanks
MJ10 commented
Hi, thanks for letting us know! You can use the utils.py
from infill_subj_arithmetic
. We'll update the repo to fix this discrepancy.
Here is the checkpoint of the fine-tuned model used as the base model for the story infilling experiments: https://drive.google.com/drive/folders/1eJibT9MPoJ1YtAOJ99QTvDQGlyc3m59H?usp=sharing