from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dialogpt.large", device="cuda")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="wizard_of_wikipedia.end2end_generator",
device="cuda",
gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dodecathlon.convai2", device="cuda", gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dialogpt.medium", device="cuda")