Enformer loaded from checkpoint does not work because of "missing positional argument: is_training"

I ran the checkpoint loading portion of the `enformer-training.ipynb` provided (I believe) by Kyle Taylor and @alimuldal, but the model cannot run a forward pass. It says `TypeError: __call__() missing 1 required positional argument: 'is_training'` even though `is_training=False` is explicitly provided as a keyword argument. Is there a workaround/fix for this? I would like to be able to access the `.trunk` attribute to compute internal embeddings of sequences, but I get the same problem when I try to call that method as well.