两个月前，我在Kaggle上训练了一个扩散模型。我将Unet2Dmodel作为pickle文件存储在一个检查点，以便以后可以加载它进行推理或在该检查点继续训练。

def save_model(model):
  with open(f'{config.output_dir}/model.pkl','wb') as f:
    pickle.dump(model,f)

def load_model(path):
  with open(path,'rb') as f:
    return pickle.load(f)

2个月前，这段代码在Kaggle和Google Colab上运行良好，但当我今天试图加载我的model.pkl文件时，我得到了这个错误：AttributeError: Can't get attribute 'AttentionBlock' on <module 'diffusers.models.attention' from '/opt/conda/lib/python3.7/site-packages/diffusers/models/attention.py'>
谢谢你，
我已经尝试改变扩散库的版本，我用2个月前，但没有变化。