search

Shark-NLP / CoNT

[NeurIPS'22 Spotlight] Data and code for our paper CoNT: Contrastive Neural Text Generation
https://arxiv.org/pdf/2205.14690.pdf
150 stars 15 forks source link

一个读取模型部分的小bug #6

Closed Luoyang144 closed 1 year ago

Luoyang144 commented 1 year ago

https://github.com/Shark-NLP/CoNT/blob/959a262ef83af00049bd6adc67b296fdd85ae6e1/model/model.py#L17 这一行的条件语句有一点点问题，应该改成elif self.PTM == "t5" or self.PTM == "codet5":.

ChenxinAn-fdu commented 1 year ago

谢谢 已经修正