paddle 2.0.2 hub.Module(name="ernie_gen_couplet")失败

[reddingit]

---

ValueError Traceback (most recent call last) /tmp/ipykernel_2835/1749205360.py in ----> 1 module = hub.Module(name="ernie_gen_couplet") 2 3 test_texts = ["人增福寿年增岁", "风吹云乱天垂泪"] 4 results = module.generate(texts=test_texts, use_gpu=True, beam_width=5) 5 for result in results:

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlehub/module/module.py in new(cls, name, directory, version, source, update, branch, ignore_env_mismatch, **kwargs) 393 branch=branch, 394 ignore_env_mismatch=ignore_env_mismatch, --> 395 **kwargs) 396 CacheUpdater("update_cache", module=name, version=version).start() 397 elif directory:

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlehub/module/module.py in init_with_name(cls, name, version, source, update, branch, ignore_env_mismatch, **kwargs) 509 510 user_module_cls.directory = directory --> 511 return user_module_cls(**kwargs) 512 513 @classmethod

~/.paddlehub/modules/ernie_gen_couplet/module.py in init(self) 46 assets_path = os.path.join(self.directory, "assets") 47 gen_checkpoint_path = os.path.join(assets_path, "ernie_gen_couplet.pdparams") ---> 48 self.model = ErnieForGeneration.from_pretrained("ernie-1.0") 49 model_state = paddle.load(gen_checkpoint_path) 50 self.model.set_dict(model_state)

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlenlp/transformers/ernie_gen/modeling.py in from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs) 370 k) 371 m[k] = v # FIXME: no need to do this in the future --> 372 model.set_state_dict(m) 373 else: 374 raise ValueError('weight file not found in pretrain dir: %s' %

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/framework.py in wrapper(*args, **kwargs) 270 kwargs['state_dict'] = kwargs['stat_dict'] 271 kwargs.pop('stat_dict') --> 272 return func(*args, **kwargs) 273 274 return wrapper

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py in set_state_dict(self, state_dict, include_sublayers, use_structured_name) 1331 global_scope(), executor) 1332 for param, state in matched_param_state: -> 1333 _set_var(param, state) 1334 1335 # [aliases] Compatible with old method names

/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py in _set_var(var, ndarray) 1323 p.set_place(t._place()) 1324 place = core.CUDAPlace(p.gpu_device_id()) -> 1325 t.set(ndarray, place) 1326 1327 executor = Executor(_get_device())._default_executor

ValueError: (InvalidArgument) Input object type error or incompatible array data type. tensor.set() supports array with bool, float16, float32, float64, int8, int16, int32, int64, uint8 or uint16, please check your input or input array data type. (at /paddle/paddle/fluid/pybind/tensor_py.h:355)