from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="gptneo.xlarge", device="cuda")
# from openchat.openchat import OpenChat # from demo.web_demo_env import WebDemoEnv from openchat import OpenChat from demo import WebDemoEnv # OpenChat(model='vqa_model_lxmert', env=WebDemoEnv()) OpenChat(model='vqa_model_oscar', env=WebDemoEnv()) # OpenChat(model='vqa_model_vilbert', env=WebDemoEnv()) # OpenChat(model='vqa_model_vinvl', env=WebDemoEnv()) # OpenChat(model='vqa_model_devlbert', env=WebDemoEnv()) # OpenChat(model='vqa_model_uniter', env=WebDemoEnv())
print("User created")
return {"success": 1}
@self.app.route('/send/<user_id>/<text>', methods=['GET'])
def send(user_id, text: str) -> Dict[str, str]:
if (self.is_empty(user_id)):
self.clear_histories(user_id)
if text in self.keywords:
# Format of self.keywords dictionary
# self.keywords['/exit'] = (exit_function, 'good bye.')
_out = self.keywords[text][1]
# text to print when keyword triggered
self.keywords[text][0](user_id, text)
# function to operate when keyword triggered
else:
model_input = self.make_model_input(user_id, text, model)
self.add_user_message(user_id, text)
_out = model.predict(model_input)
self.add_bot_message(user_id, _out['output'])
return {"output": _out['output']}
if __name__ == '__main__':
self.app.run()
from openchat import OpenChat
OpenChat(model="blender.small",
env=WebDemoEnv(),
device='cpu',
environment="webserver")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="reddit.xlarge", device="cuda")
from openchat import OpenChat OpenChat(model="blender.medium", device="cpu")
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="blender.xxlarge", device="cuda", gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dodecathlon.daily_dialog", device="cuda", gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="gptneo.small", device="cuda")
from openchat import OpenChat OpenChat(model="dialogpt", size="large")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dodecathlon.wizard_of_wikipedia", device="cuda", gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="unlikelihood.convai2.vocab.alpha.1e-1",
device="cuda",
gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dialogpt.large", device="cuda")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dodecathlon.convai2", device="cuda", gpu=0)
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dodecathlon.all_tasks_mt", device="cuda")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="blender.small", device="cuda", gpu=0)
import os
import sys
sys.path.append(os.path.dirname(os.path.abspath(os.path.dirname(__file__))))
from openchat import OpenChat
from web_demo_env import WebDemoEnv
# CPU Running
print('cpu')
OpenChat(model="blenderbot", size="large", env=WebDemoEnv())
# GPU Running
#print('gpu')
#OpenChat(model="blenderbot", size="large", env=WebDemoEnv(), device="cuda")
from openchat import OpenChat
if __name__ == '__main__':
OpenChat(model="dialogpt.medium", device="cuda")