Fix plato-2 and plato-mini dtype bug (#767)

* fix unified transformer dtype problem

* fix win dtype bug

* Fix plato-2 and plato-mini dtype bug

Author: smallv0221

examples/dialogue/plato-2/interaction.py

@@ -76,6 +76,7 @@ def interact(args):
                 example, is_infer=True)
             data = plato_reader._pad_batch_records([record], is_infer=True)
             inputs = gen_inputs(data, args.latent_type_size)
+            inputs['tgt_ids'] = inputs['tgt_ids'].astype('int64')
             pred = model(inputs)[0]
             bot_response = pred["response"]
             print(

examples/dialogue/unified_transformer/interaction.py

@@ -48,6 +48,7 @@ def interaction(args, model, tokenizer):
                 add_start_token_as_response=True,
                 return_tensors=True,
                 is_split_into_words=False)
+            inputs['input_ids'] = inputs['input_ids'].astype('int64')
             ids, scores = model.generate(
                 input_ids=inputs['input_ids'],
                 token_type_ids=inputs['token_type_ids'],