尝试运行gpt2的生成函数时tensorflow中出现不兼容的形状错误？

Question

我已经在colab上使用自己的数据集对gpt-2进行了微调，并且我成功地生成了文本，但是当我下载了我的模型并尝试在spyder中生成文本时，我得到了这个值错误：尝试共享变量模型/wpe，但指定了形状（10241280）并找到了形状（1024768）

这是我运行的代码` 将gpt_2_简单导入为gpt2

          sess = gpt2.start_tf_sess()
          gpt2.load_gpt2(sess, run_name='path to model file')
          gpt2.generate(sess,
          temperature=0.7,
          top_k=40,
          nsamples=1,
          batch_size=1,
          length=200,
          sample_delim=''
          )` 

以下是错误：

` 文件“”，第11行，在 样本_delim=“”

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\gpt_2\u simple\gpt_2.py”，第459行，在generate中 温度=温度，top_k=top_k，top_p=top_p

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\gpt\u 2\u simple\src\sample.py”，第67行，按样本顺序 上下文\输出=步骤（hparams，上下文[：，：-1]）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\gpt\u 2\u simple\src\sample.py”，第52行，步骤 过去=过去，重用=tf.compat.v1.AUTO\u重用）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\gpt\u 2\u simple\src\model.py”，模型中第183行 初始化器=tf.compat.v1.random\u normal\u初始化器（stddev=0.01））

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\tensorflow\u core\python\ops\variable\u scope.py”，第1501行，在get\u variable中 聚合=聚合）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\tensorflow\u core\python\ops\variable\u scope.py”，第1244行，在get\u variable中 聚合=聚合）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\tensorflow\u core\python\ops\variable\u scope.py”，第567行，在get\u variable中 聚合=聚合）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\tensorflow\u core\python\ops\variable\u scope.py”，第519行，在\u true\u getter中 聚合=聚合）

文件“C:\Users\Wahiba\anaconda3\envs\PythonKerasCPU\lib\site packages\tensorflow\u core\python\ops\variable\u scope.py”，第874行，在“get\u single\u variable”中 （名称、形状、已找到值\u var.get\u形状（））

ValueError:尝试共享变量模型/wpe，但指定了形状（10241280）并找到了形状（1024768）。`

enter image description hereenter image description here

我做错了什么