a error when I run sample.py

[beautyMJ]

When I run the sample.py, an error occurred:
Traceback (most recent call last):
File "sample.py", line 57, in
model = Model(saved_args, True)
File "C:\YMJ\download_pro\magenta-master\magenta\models\sketch_rnn\sketch-rnn-master\model.py", line 93, in init
outputs, states = custom_rnn_autodecoder(inputs, self.initial_input, self.initial_state, cell, scope='rnn_mdn')
File "C:\YMJ\download_pro\magenta-master\magenta\models\sketch_rnn\sketch-rnn-master\model.py", line 71, in custom_rnn_autodecoder
output, new_state = cell(inp, states[-1])
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 183, in call
return super(RNNCell, self).call(inputs, state)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\layers\base.py", line 575, in call
outputs = self.call(inputs, *args, **kwargs)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 1066, in call
cur_inp, new_state = cell(cur_inp, cur_state)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 183, in call
return super(RNNCell, self).call(inputs, state)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\layers\base.py", line 575, in call
outputs = self.call(inputs, *args, **kwargs)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 441, in call
value=self._linear([inputs, h]), num_or_size_splits=4, axis=1)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\rnn_cell_impl.py", line 1189, in call
res = math_ops.matmul(array_ops.concat(args, 1), self._weights)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\math_ops.py", line 1891, in matmul
a, b, transpose_a=transpose_a, transpose_b=transpose_b, name=name)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\ops\gen_math_ops.py", line 2436, in _mat_mul
name=name)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2958, in create_op
set_shapes_for_outputs(ret)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2209, in set_shapes_for_outputs
shapes = shape_func(op)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\ops.py", line 2159, in call_with_requiring
return call_cpp_shape_fn(op, require_shape_fn=True)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\common_shapes.py", line 627, in call_cpp_shape_fn
require_shape_fn)
File "C:\ProgramData\Anaconda3\envs\magenta\lib\site-packages\tensorflow\python\framework\common_shapes.py", line 691, in _call_cpp_shape_fn_impl
raise ValueError(err.message)
ValueError: Dimensions must be equal, but are 512 and 261 for 'rnn_mdn_1/rnn_mdn/multi_rnn_cell/cell_0/cell_0/basic_lstm_cell/MatMul_1' (op: 'MatMul') with input shapes: [1,512], [261,1024].
Could you tell me how to solve it? Thank you very much!