示例#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
from openchat import OpenChat

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