コード例 #1
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="dialogpt.large", device="cuda")
コード例 #2
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="wizard_of_wikipedia.end2end_generator",
             device="cuda",
             gpu=0)
コード例 #3
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="dodecathlon.convai2", device="cuda", gpu=0)
コード例 #4
0
ファイル: dialogpt_medium.py プロジェクト: ibivibiv/openchat 
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="dialogpt.medium", device="cuda")