Exemplo n.º 1
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="dialogpt.large", device="cuda")
Exemplo n.º 2
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="wizard_of_wikipedia.end2end_generator",
             device="cuda",
             gpu=0)
Exemplo n.º 3
0
from openchat import OpenChat

if __name__ == '__main__':
    OpenChat(model="dodecathlon.convai2", device="cuda", gpu=0)
Exemplo n.º 4
0
from openchat import OpenChat

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