Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fix dialogue tutorial bug (NVIDIA#5297) (NVIDIA#5303)
* set add_pooling_layer=False for huggingface bert model * remove add_pooling_layer=False and set find_unused_parameters=True * set num_prompt_tokens to 0 for huggingface Co-authored-by: Zhilin Wang <wangzhilin12061996@hotmail.com> Co-authored-by: Eric Harper <complex451@gmail.com>
- Loading branch information